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]: 降低一维对应输出