人脸识别:如何识别未知人脸?
在代码中没有处理未知人脸的逻辑,因此没有人被识别为unknown。可以在代码中加入对未知人脸的处理逻辑,例如设置一个阈值,当模型的预测得分低于阈值时,将其识别为unknown。可以通过以下代码实现:
threshold = 0.8 # 设置阈值
output = network(face)
predicted_class = np.argmax(output.asnumpy(), axis=1)
if output.asnumpy()[0][predicted_class[0]] < threshold:
label = 'unknown'
else:
label = subjects[predicted_class[0]]
cv2.putText(img1, label, (x, y), cv2.FONT_HERSHEY_COMPLEX, 1, (128, 128, 0), 2)
这样,当模型的预测得分低于设定的阈值时,就会将其识别为unknown。可以根据实际情况调整阈值的大小。
原文地址: https://www.cveoy.top/t/topic/jqCa 著作权归作者所有。请勿转载和采集!