保存裁剪的邊界框

Question

我想從包含多類的所有數據集中裁剪和保存圖像。 但沒有 output。 謝謝。

def detect_face(img):
    detector = MTCNN()
    faces = detector.detect_faces(image) 
return faces     
                                                                               
dataset = '/content/drive/MyDrive/data_LFW/'

for img in glob.glob(dataset+'/*.*'):
    var_img = cv2.imread(img)
    face = detect_face(var_img)

    if (len(face) == 0):
       continue

    for(ex, ey, ew, eh) in face:
        crop_image = var_img[ey:ey+eh, ex:ex+ew]
                                                                                                           
     
   cv2.imwrite(os.path.join("/content/drive/MyDrive/data_crop",str(img)) ,crop_image)

Answer 1

假設其他一切都是正確的（因為我沒有你的數據集和檢測器代碼），你可能想嘗試糾正幾個縮進，看看你是否能得到一些不同的結果：

if (len(face) == 0):
    continue #correct the indentation and try again

如果您想保存每個檢測，請更正 cv2.imwrite 縮進：

for img in glob.glob(dataset+'/*.*'):
    var_img = cv2.imread(img)
    face = detect_face(var_img)

    if (len(face) == 0):
        continue

    for(ex, ey, ew, eh) in face:
        crop_image = var_img[ey:ey+eh, ex:ex+ew]
      
    cv2.imwrite(os.path.join("/content/drive/MyDrive/data_crop",str(img)),crop_image)