NotADirectoryError: [Errno 20] 不是目录: 'known_faces/.DS_Store'

如何解决NotADirectoryError: [Errno 20] 不是目录: 'known_faces/.DS_Store'

如何修复以下错误：

NotADirectoryError: [Errno 20] 不是目录： 'known_faces/.DS_Store'

代码

import face_recognition
import os
import cv2
import numpy as np

KNOWN_FACES = "known_faces"
UNKNOWN_FACES = "unknown_faces"
TOLERANCE = 0.6
THICKNESS = 3
MODEL = "cnn"

known_faces = []
known_names = []

for name in os.listdir(KNOWN_FACES):
    for filename in os.listdir(f"{KNOWN_FACES}/{name}"):

        image = face_recognition.load_image_file(f"{KNOWN_FACES}/{name}/{filename}")
        encoding = face_recognition.face_encodings(image)
        known_faces.append(encoding)
        known_names.append(name)

print("processing unknown_faces")
for filename in os.listdir(UNKNOWN_FACES):
    print(filename)

    image = face_recognition.load_image_file(f"{UNKNOWN_FACES}/{filename}")
    locations = face_recognition.face_locations(image,model=MODEL)
    encodings = face_recognition.face_encodings(image,locations)
    image = cv2.cvtColor(image,cv2.COLOR_RGB2BGR)

    for face_encoding,face_locations in zip(encodings,locations):
        
        results = face_recognition.compare_faces(face_encoding,known_faces,TOLERANCE)
        MATCH = None
        if True in results:
            match = known_names[results.index(True)]
            print(f"Match found: {match}")

            top_left = (face_location[3],face_location[0])
            bot_right = (face_location[1],face_location[2])

            color = [0,255,0]

            cv2.rectangle(image,top_left,bot_right,color,THICKNESS)

            top_left = (face_location[3],face_location[2] + 22)
            cv2.rectangle(image,cv2.FILLED)
            cv2.putText(image,math,(face_location[3]+10,face_location[2])+15,cv2.FONT_HERSEY_SIMPLEX,0.5,(200,200,200),THICKNESS)

    cv2.imshow(filename,image)
    cv2.waitKey(10000)

解决方法

os.listdir(KNOWN_FACES) 返回 KNOWN_FACES 目录中的所有文件。在您的特定情况下，还有 .DS_Store 文件。

您可以仅考虑目录并排除 .DS_Store 等文件来过滤结果。

import os

for name in os.listdir(KNOWN_FACES):
    dir_path = os.path.join(KNOWN_FACES,name)
    # if it's a directory
    if os.path.isdir(dir_path):
       for filename in os.listdir(dir_path):
          # if the file is a valid file (a better way could be to check your specific extension,e.g.,png)
          if not filename.startswith('.'): 
              filepath = os.path.join(dir_path,filename)
              image = face_recognition.load_image_file(filepath)

,

在您的“known_faces”目录中有一个名为 .DS_Store 的文件。由于您只想查看“known_faces”目录中的目录，因此需要删除该文件。

正如 abc 所指出的，您可能只想通过使用 os.path.isdir() 来检查您正在查找的目录。