opencvとPythonによる顔画像の切り取り

Question

実現したいこと

python初心者です。
初歩的な質問ですみません。
pythonで顔認識用のデータを集めるためのプログラムを作成しています。
opencvを用いた顔画像データの切り取りをする機能を実装中に以下のエラーメッセージが発生しました。

発生しているエラーメッセージ

  File "face2.py", line 21, in <module>
cv2.error: OpenCV(4.1.1) C:\projects\opencv-python\opencv\modules\imgproc\src\color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cv::cvtColor'

該当のソースコード

# -*- coding:utf-8 -*-
import cv2
import numpy as np

# 集めてきた画像データのあるディレクトリ
input_data_path = (r'C:/Users/python/img/ayami-nakajyo')
# 切り抜いた画像の保存先ディレクトリ
save_path = (r'C:/Users//python/img/cut-nakajyo')
# OpenCVのデフォルトの分類器のpath。
cascade_path = 'C:/Users/Anaconda3/Lib/site-packages/cv2/data/haarcascade_frontalface_default.xml'
faceCascade = cv2.CascadeClassifier(cascade_path)

# 収集した画像の枚数
image_count = 106
# 顔検知に成功した数
face_detect_count = 0

# 集めた画像データから顔が検知されたら、切り取り、保存する。
for i in range(image_count):
  img = cv2.imread(r'img_' + str(i) + '.jpg', cv2.IMREAD_COLOR)
  gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
  face = faceCascade.detectMultiScale(gray, 1.1, 3)
  if len(face) > 0:
    for rect in face:
      # cv2.rectangle(img, tuple(rect[0:2]), tuple(rect[0:2]+rect[2:4]), (0, 0,255), thickness=1)
      # cv2.imwrite('detected.jpg', img)
      x = rect[0]
      y = rect[1]
      w = rect[2]
      h = rect[3]
      cv2.imwrite(save_path + 'cut' + str(face_detect_count) + '.jpg', img[y:y+h, x:x+w])
      face_detect_count = face_detect_count + 1
  else:
    print ('image' + str(i) + ':NoFace')

試したこと

カスケードファイルのパスを確認しましたが、パスの場所は間違っていませんでした。
読み込み画像のファイルの名前も確認しましたが、該当する名前（img_1.jpg、img_2.jpg…img_106.jpg）で存在しています。
windows環境はrow文字に直さないといけないとあったので、ｒを追加しましたが、相変わらず同じエラーが出力されます。

補足情報（FW/ツールのバージョンなど）

Answer 1

パスが間違っており、読み込みに失敗していると思います。
input_data_path という変数はどこにも使用されていないようですが、この場所に画像があるのでしたら、os.path.join() でパスを結合する必要があると思います。

img = cv2.imread(os.path.join(input_data_path, r'img_' + str(i) + '.jpg'), cv2.IMREAD_COLOR)

Comments

[as15]

tiitoiさん、ご回答ありがとうございます。
img = cv2.imread(r'img_' + str(i) + '.jpg', cv2.IMREAD_COLOR)
の部分を
img = cv2.imread(os.path.join(input_data_path, r'img_' + str(i) + '.jpg'), cv2.IMREAD_COLOR)
に変更し、実行した所、
File "face3.py", line 20, in <module>
NameError: name 'os' is not defined
と出力されました。
そのため、初めの文にimport os　の一文を追加したのですが、
また同じエラーが出力されてしまいました…