all_boxes,landmarks = mtcnn_detector.detect_face(test_data)
count = 0
print(all_boxes)
print(landmarks)
for imagepath in gt_imdb:
image = cv2.imread(imagepath)
for bbox in all_boxes:
# cv2.putText(image,str(np.round(bbox[4],2)),(int(bbox[0]),int(bbox[1])),cv2.FONT_HERSHEY_TRIPLEX,1,color=(255,0,255))
print("%d %d %d %d" %(bbox[0],bbox[1],bbox[2],bbox[3]))
cv2.rectangle(image, (int(bbox[0]),int(bbox[1])),(int(bbox[2]),int(bbox[3])),(0,0,255), 1)
代码中all_boxes 是一个三维 的numpy.array 类型
所以需要把 for bbox in all_boxes:
改成 for bbox in all_boxes[count]: 降低一维对应输出
本文详细介绍使用MTCNN进行人脸检测的过程,包括加载数据、调用MTCNN模型进行人脸检测并绘制检测框,同时解析了all_boxes的维度问题及解决方法。
3万+

被折叠的 条评论
为什么被折叠?



