[python] 얼굴검출되는 이미지와 검출되지 않는 이미지 분류해서 폴더별로 저장시키기

여러 이미지중에

얼굴검출되는 이미지와 검출되지 않는 이미지 분류해서 폴더별로 저장시키기

 

 

 

import face_recognition
from matplotlib import pyplot as plt
import cv2
from PIL import Image
import os
import shutil

#확인 할 이미지가 들어있는 폴더
imagepath = 'D:/my/video frame/cropface/'
#인식 안되는 얼굴들 이동할 폴더 경로
mvpath = 'D:/my/video frame/noface2/'


for root, dirs, files in os.walk(imagepath):
    #print('root > ',root, 'dirs >',dirs,'files >', files)
    for idx, file in enumerate(files):
        #print("index : {}, value: {}".format(idx,file))
        #print('idx > ',idx,' file > ', file)        
        imgfile = imagepath+file
        face_locations = 0
        image = face_recognition.load_image_file(imgfile) #, mode='RGB'
        face_locations = face_recognition.face_locations(image)
        #print("I found {} face(s) in this photograph.".format(len(face_locations)))
        
        if len(face_locations) == 0:
            print(idx,'>> ',imagepath+file)
            shutil.move(imagepath+file, mvpath+file)
            #print(imagepath+file)
        else:
            print(idx,'>> ',len(face_locations))
    print('finish')