フォルダ内画像一括処理

Question

https://www.robotech-note.com/entry/2016/12/08/204143
ここを元に、フォルダ内画像一括処理を試みようとしました。

# -*- coding: utf-8 -*-
import os
import cv2

def main():
    data_dir_path = u"./画像ディレクトリ名/"
    file_list = os.listdir(r'./画像ディレクトリ名/')

    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
            image = cv.imread(abs_name)
            #以下各画像に対する処理を記載する

if __name__ == '__main__':
    main()

「#以下各画像に対する処理を記載する」の直下に、ある画像に関する処理を書き込めばよいのだと思うのですが、この「ある画像」のファイル名は何なんでしょうか？
「u」でも「u.png」でも「file_name」でも「file_name.png」でもなかった。

特定の画像を二値化するプログラムを作り、実際に動いたが、複数の画像に対し行う必要があるため、1つ1つやるのは大変であるから今度は一括処理をしたいと思ったのですが、

cv2.error: OpenCV(4.5.1) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-wvn_it83\opencv\modules\core\src\arithm.cpp:1743: error: (-215:Assertion failed) ! _src.empty() in function 'cv::inRange'

という、ファイル名に起因？するエラーが出ます。

ちなみにとりあえず作ったプログラム（エラーが生じる）は、

import os
import cv2

os.chdir('C://Users//username//Desktop//cv')

def main():
    data_dir_path = u"./ok/"
    file_list = os.listdir(r'./ok3/')

    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
            image = cv.imread(abs_name)
            #以下各画像に対する処理を記載する

img = cv2.imread('file_name')  # 画像の読み出し
img = cv2.inRange(img, (120, 21, 145), (214,152,204))

cv2.imwrite('nitikafile_name',img)  # 画像の保存 

if __name__ == '__main__':
    main()

になりまう、ごく初歩的なところで引っかかっている気がします。

Answer 1

for file_name in file_list:  # ファイル名についてのループ
        root, ext = os.path.splitext(file_name) # extに拡張子を取り出し
        if ext == u'.png' or u'.jpeg' or u'.jpg': # 拡張子が画像ならば以下の処理
            abs_name = data_dir_path + '/' + file_name # ファイルのパスをabs_nameに格納
            image = cv.imread(abs_name) # abs_nameのファイルをcv.imreadで読み込み
            #以下各画像に対する処理を記載する

となっています。したがって、読み込まれた画像はimageに格納されていると思います。

Comments

[退会済みユーザー]

import os
import cv2

os.chdir('C://Users//username//Desktop//cv')

def main():
data_dir_path = u"./ok/"
file_list = os.listdir(r'./ok3/')

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
image = cv.imread(abs_name)
#以下各画像に対する処理を記載する

img = cv2.inRange(image, (120, 21, 145), (214,152,204))

cv2.imwrite('nitikafile_name',img) # 画像の保存

if __name__ == '__main__':
main()

こう書いてみたのですが、

>>> img = cv2.inRange(image, (120, 21, 145), (214,152,204))
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
NameError: name 'image' is not defined
>>>
>>> cv2.imwrite('nitikafile_name',img) # 画像の保存
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
cv2.error: OpenCV(4.5.1) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-wvn_it83\opencv\modules\imgcodecs\src\loadsave.cpp:753: error: (-215:Assertion failed) !_img.empty() in function 'cv::imwrite'

と出ますね。

[Amakaze]

imageに何も格納されていないとすると、forループの中で画像の読み込みができていないように思えます。

image = cv.imread(abs_name)の前後で適当にprintなどしてみて、そもそも、そこの処理に突入できているか確認してみてください。もしできていなければ、フォルダのパスを確認してみてください。

[退会済みユーザー]

import os
import cv2

os.chdir('C://Users//username//Desktop//cv')

def main():
data_dir_path = u"./ok/"
file_list = os.listdir(r'./ok3/')

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

print('出力したい文字列')

とすると、出力したい文字列が表示されますね。

[退会済みユーザー]

ただなんか無茶苦茶エラー出てますね

>>> for file_name in file_list:
File "<stdin>", line 1
for file_name in file_list:
IndentationError: unexpected indent
>>> root, ext = os.path.splitext(file_name)
File "<stdin>", line 1
root, ext = os.path.splitext(file_name)
IndentationError: unexpected indent
>>> if ext == u'.png' or u'.jpeg' or u'.jpg':
File "<stdin>", line 1
if ext == u'.png' or u'.jpeg' or u'.jpg':
IndentationError: unexpected indent
>>> abs_name = data_dir_path + '/' + file_name
File "<stdin>", line 1
abs_name = data_dir_path + '/' + file_name
IndentationError: unexpected indent

[Amakaze]

data_dir_path = u"./ok/"
file_list = os.listdir(r'./ok3/')
ここのフォルダ名が異なっているので、画像が正しく読めていないのではないでしょうか。

エラーについてはインデントをちゃんと合わせれば消えると思います。

[退会済みユーザー]

色々変えてみたのですが（作業フォルダは初期フォルダにしました）

import os
import cv2

def main():
data_dir_path = u"./ok/"
file_list = os.listdir(r'./ok/')
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
image = cv.imread(abs_name)
#以下各画像に対する処理を記載する

img = cv2.inRange(image, (120, 21, 145), (214,152,204))

cv2.imwrite('21nitika.png',img) # 画像の保存

if __name__ == '__main__':
main()

とすると、
>>> img = cv2.inRange(image, (120, 21, 145), (214,152,204))
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
NameError: name 'image' is not defined
>>>
>>> cv2.imwrite('21nitika.png',img) # 画像の保存
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
cv2.error: OpenCV(4.5.1) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-wvn_it83\opencv\modules\imgcodecs\src\loadsave.cpp:753: error: (-215:Assertion failed) !_img.empty() in function 'cv::imwrite'
というエラーが出ます。

Answer 2

処理する対象の場所には、abs_nameと書けば良いのです。

pythonでのいわゆる変数というものを理解してください。