Python OpenCVリテラルエラーが無効

Question

私はValueError: invalid literal for int() with base 10: '.DS'エラーがあります。私は一日の解決策を探していますが、解決できませんでした。私のディレクトリは空ではありません。私はOpenCVを使ってfacerecognizerを訓練しようとしています。私はtrainディレクトリから顔の写真を取得しようとしていますが、私はできません。私のコード

：電車の中で

images = [] 
labels = [] 
for file in os.listdir('train/'): 
    image = cv2.imread('train/'+file, 0) 
    images.append(image) 
    labels.append(int(file.split('_')[0])) 

写真に名前が付いているので上の1_001.pmg、1_002.pmgとあります。私は愚かな問題のためにあなたの助けを求めています。ありがとう！

ps。あなたが必要とする、このコードはここで@Fabianの答えhow to notify user if there are common faces detected between two directories with python and opencv

Answer 1

それはあなたに隠すことができる指定されたディレクトリ内のすべてのファイルのリストを返すような問題は、os.listdir('train/')であるからである、と'.DS_Store'は、それらのファイルの中にもあります戻り値の文字列が.pngであるかどうかを確認してから、cv::imreadに渡します。

images = [] 
labels = [] 
for file_name in os.listdir('train/'): 
    if file_name.endswith(".png"): 
     image = cv2.imread('train/'+file_name, 0) 
     images.append(image) 
     labels.append(int(file_name.split('_')[0]))