OpenCV（python3）を用いたフォルダ内画像の一括フィルタ加工に関するご質問です

Question

前提・実現したいこと

OpenCV, pythonにて画像編集を行おうとしています。
あるフォルダ内の画像全てに対してフィルター処理などを行おうとしておりますが、
その際次のようなエラーメッセージが発生しました。

なお、ソースコードはガウシアンフィルタの例です。
「before」フォルダ内の画像を全て読み込み、フィルタ処理を行ったのちに「after」フォルダに保存しようとしています。

プログラミング初心者のため、基本的なことが原因でしたらすみません。
アドバイスのほどよろしくお願いします。

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

error: OpenCV(4.1.2) /Users/travis/build/skvark/opencv-python/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

該当のソースコード

import os
import cv2
import numpy as np

def main():
    data_dir_path = u"./before/"
    file_list = os.listdir(r'./before/')
    data_dir_path2 = u"./after/"

    for file_name in file_list:
        root, ext = os.path.splitext(file_name)
        if ext == u'.png' or u'.jpeg' or u'.jpg':
            abs_name = data_dir_path + '/' + file_name
            img = cv2.imread(abs_name)
            #以下各画像に対する処理を記載する
            gray_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
            dst = cv2.GaussianBlur(gray, ksize=(3, 3), sigmaX=1.3)
            abs_name2 = data_dir_path2 + '/' + file_name
            cv2.imwrite(abs_name2, dst)

if __name__ == '__main__':
    main()

試したこと

フィルタ処理に関しては１画像のみの処理ではうまく行えていたのですが、複数一括処理を行うところでうまく行きません。
エラーから推測するに、うまく画像が取り込めていないのではと考えますが、どこが悪いのかどう対応すれば良いかがわかりません。

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

macOS Catalina(10.15.5)
jupyter notebook6.0.1

Answer 1

Python 2 をお使いなのでしょうか？
補足情報に Python のバージョンもあったほうがよいと思います。

問題点

以下の条件式は意図どおり動きません。

if ext == u'.png' or u'.jpeg' or u'.jpg':

「ext == u'.png' または ext == u'.jpeg' または ext == u'.jpg'」 ではなく、「ext == u'.png' または u'.jpeg' または u'.jpg'」 と解釈されます。空でない文字列は True と評価されるので、したがって、この条件式は常に True になります。
以下のように修正するとよいでしょう。

if ext == u'.png' or ext == u'.jpeg' or ext == u'.jpg':
または
if ext in [u'.png', u'.jpeg', u'.jpg']:

補足

error: OpenCV(4.1.2) /Users/travis/build/skvark/opencv-python/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

cv2.imread() はファイルを読み込みに失敗した場合は None を返し、その時点ではエラーにならず、後続の処理 cv2.cvtColor() でエラーになります。
ファイル読み込みに失敗するときに気づけるように、img が None でないことをチェックする assert を入れておくとよいと思います。

img = cv2.imread(abs_name)
assert img is not None, os.path.abspath(abs_name)

Comments

[MT23]

詳しくないものにもわかりやすくご説明頂きありがとうございました。
解決いたしました。
assertにつきましては早速追加してみました。
なるほどですね。
なお、pythonのバージョンはpython3でした。
以後気をつけます。

[tiitoi]

すいません。タイトルに Python3 と書いてありましたね。失礼しました。
解決したようでよかったです。