!_src.empty() in function 'cvtColor'のエラー

Question

pythonにてマルチラベル分類を行った際、下のようなエラーが出ました。

cv2.error: OpenCV(4.5.3) /tmp/pip-req-build-s58vuwxh/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

このエラーの原因となるコードは

    def __getitem__(self, index):
        image = cv2.imread(f"../input/movie-classifier/Multi_Label_dataset/Images/{self.image_names[index]}.png")
        
        # convert the image from BGR to RGB color format
        image = cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
     
        # apply image transforms
        image = self.transform(image)
        targets = self.labels[index]
        
        return {
            'image': torch.tensor(image, dtype=torch.float32),
            'label': torch.tensor(targets, dtype=torch.float32)
        }

RGBに変換する際にエラーが出たと考えられます。
データセットにはグレースケール画像はないため,もともとRGBの画像が含まれていることが原因でエラーが出ているのでしょうか？
その際RGBの画像を見分ける方法を教えてくださると助かります。

ちなみに途中まで画像の読み込みはできているので一部の画像がエラーの原因になっていると思います。。。

    def __getitem__(self, index):
        image = cv2.imread(f"../input/movie-classifier/Multi_Label_dataset/Images/{self.image_names[index]}.png")
        print(image.shape)

        # convert the image from BGR to RGB color format
        image = cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
     
        # apply image transforms
        image = self.transform(image)
        targets = self.labels[index]
        
        return {
            'image': torch.tensor(image, dtype=torch.float32),
            'label': torch.tensor(targets, dtype=torch.float32)
        }

上のようにプリント文で書き出したところ、途中まで読み込めていましたが
途中で

'NoneType' object has no attribute 'shape'

というエラーが出てました。
この画像を調べる方法を教えてくださると助かります。。

またフォルダ内のshapeを全て調べたところエラーは１枚も出ていないため、何枚か読み込みに失敗していると考えられるのですが、全ての画像が同フォルダ内にあるとき、エラーの原因は何でしょうか。

Comments

[fana]

何故正当な正否判定処理を頑なに設けないのですか？
ふつーに結果をチェックして失敗と判定できたならば，そのときにimreadに与えたパス文字列を表示なりなんなりしてみたら良いのではないですか？

[fana]

Why do you hate implementing cv2.imread result checking?
Why do you refuse to identify the name of the file that failed to be read?

Answer 1

_src.empty() in function 'cvtColor'

画像が読み込めていなくて、空っぽという意味です。
値をプリントしてみてください。

>>> image = cv2.imread('none.jpg')
>>> print(image)
None

>>> im2 = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
cv2.error: OpenCV(4.5.3) color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

参考ですが、
cv2.imreadで画像が見つからないことはよくあります。
以下のようにtry-exceptでエラーチェックするのが良いでしょう。

try:
    img = cv2.imread(画像ファイル名)
    if img is None:
        raise FileNotFoundError('ファイルが見つかりません')

except FileNotFoundError as e:
    print(e)

Answer 2

image = cv2.imread( ...

これが失敗しているだけの話ではないでしょうか．

というか，ファイル読込という処理は普通に失敗があり得るのだから，次の処理に進む前に読込成否のチェックを入れるべき．
→で，それで読込失敗となったときのファイルパスでも確認してみれば何かわかるのでは？