画像同時大量二値化。

Question

import cv2  # OpenCVのインポート

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

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

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

上のようなプログラムを作りました、１枚の画像を二値化します、これを、

import os
import glob
from PIL import Image

img_files = glob.glob('*.png')
width = 640
height = 480

for f in img_files:
    img = Image.open(f)
    img_resize = img.resize((width,height))
    fname, fext = os.path.splitext(f)
    img_resize.save(fname + '_640x480' + fext)

上のプログラムと組み合わせ、複数画像を二値化したいのですが、

import os
import glob
from PIL import Image

img_files = glob.glob('*.png')

for f in img_files:
img = Image.open(f)
img_resize = cv2.inRange(img, (120, 21, 145), (214,152,204))
fname, fext = os.path.splitext(f)
img_resize.save(fname + '_640x480' + fext)

このように書いても、
Traceback (most recent call last):
File "<stdin>", line 3, in <module>
TypeError: Expected Ptrcv::UMat for argument 'src'
というエラーが出ます、たぶんcv2を使うあたりでおかしくなっているのですが、
どう書けばよいのかわかりません、どのようにするとうまくいくでしょうか。

Comments

[ppaul]

for f in img_files: の下の行がインデントされていませんが、これでエラーは出ていませんか？

[退会済みユーザー]

インデント系のエラーは、よく出てるので無視してます。

Answer 1

組み合わせ方が間違っています。最初のコードではcv2(OpenCV)、二番目のコードではPIL(Pillow)が使われており、それが混ざってしまっています。

そのため、Imageで読んだデータをcv2.inRangeで処理しようとして、件のエラーが発生しています。

二番目のコードからはglobを使った繰り返しだけ採用し、画像処理は最初のコードのものを使えばいい。

なお、640x480にリサイズしているわけではないので、2値化後の変数名やファイル名はそれっぽく変えた方がいいでしょう。そうでないと、後で読み直すときに混乱します。

import os
import glob
import cv2

for f in glob.glob('*.png'):
    img = cv2.imread(f)
    img_bin = cv2.inRange(img, (120, 21, 145), (214, 152, 204))
    fname, fext = os.path.splitext(f)
    cv2.imwrite(fname + '_bin' + fext, img_bin)

Comments

[退会済みユーザー]

これはすごい！本当にできましたありがとうございます、

ちなみにファイルを上書きするには、どうすればいいんでしょうか？
cv2.imwrite(fname, img_bin)
に変えると、
File "<stdin>", line 5, 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:682: error: (-2:Unspecified error) could not find a writer for the specified extension in function 'cv::imwrite_'
というエラーが・・・。

[Daregada]

上書きするなら、fnameではなくてfですよ。
cv2.imwrite(f, img_bin)

Answer 2

単に画像ファイル以外のものが混じってるってだけでは

Comments

[退会済みユーザー]

txtファイルも入れてますね、pngのみにしてみます。