Pythonによる画像仕分けプログラム

Question

簡単な画像仕分けプログラムを作成したいです。
内容は画像を表示させその画像を人間が黙視し
０、１、２キーを押したら任意のフォルダに画像が移動するプログラムを作成したいです。
動物フォルダの画像を１枚ずつ表示させ
０キーは猫フォルダ
１キーは犬フォルダ
２キーはその他フォルダに移動させ
次の画像を表示させ仕分けるを繰り返すプログラムを作成したいです。
現在このような感じですが動きません。いい方法と何がいけないのかを教えてください。
モジュールはopencvにこだわりません。

import cv2
import os
import shutil
# def load_lists(list_dir):
#     classes = list_dir.read().split()
#     list_dir.close()
#     return classes

#classes,///list
IMG_SIZE=150
link0='/Users/hoge/Downloads/猫'
link1='/Users/hoge/Downloads/犬'
link2='/Users/hoge/Downloads/’その他'
link3='/Users/hoge/Downloads/動物'
data_list = os.listdir(link3)

for data in data_list:
    if(os.path.basename(data) == '.DS_Store'):
            print('Escape')
    else:
        img = cv2.imread(data)
        img = cv2.resize(img,(IMG_SIZE,IMG_SIZE))
        img = img/255.0
        cv2.imgshow(data,img)
        if cv2.waitKey(48):
            shutil.move(link3,link0)
            cv2.destroyAllwindows()
        elif cv2.waitKey(49):
            shutil.move(link3,link1)
            cv2.destroyAllwindows()
        elif cv2.waitKey(50):
            shutil.move(link3,link2)
            cv2.destroyAllwindows()

Answer 1

とりあえず以下の部分はこうした方がいいと思います。
waitKey()の引数は待ち時間[ms]で、返り値が入力キーです。
引数を省略した場合、キー入力があるまで待ってくれます。

key = cv2.waitKey()
if key == 48:
    shutil.move(link3,link0)
elif key == 49:
    shutil.move(link3,link1)
elif key == 50:
    shutil.move(link3,link2)

---

ファイル名がそもそも正しく得られているか確認するコード

import os

link3='/Users/hoge/Downloads/動物'
data_list = os.listdir(link3)

for data in data_list:
    print(data)

---

画像を表示、入力キーを判別するまでは出来ましたよ。Windowsですが...

filenames = os.listdir(src)
for file in filenames:
    img = cv2.imread(src+'/'+file)
    if img is None:
       print('through')
    else:
       cv2.imshow("image", img)
       key = cv2.waitKey()
       
       if key == 48:
            print('0')
       elif key == 49:
            print('1')
       elif key == 50:
            print('2')
       else:
            print('pass', str(key))

Comments

[kaitokimura]

ありがとうございます。笑
ですがデータの渡し方が根本的な問題のようです。
OpenCV Error: Assertion failed (ssize.area() > 0) in resize, file /Users/jenkins/miniconda/1/x64/conda-bld/conda_1486587097465/work/opencv-3.1.0/modules/imgproc/src/imgwarp.cpp, line 3229
これは経験的に意味のない値をリサイズに与えている気がします…。

[LouiS0616]

imreadした直後にimshow, waitkeyをしてみてください。
表示できないのであれば読み取りに問題が、出来るのであればresizeの使い方に問題があることになります。

[kaitokimura]

読み取りに問題があるようです。１枚１枚画像を開きたい場合どのような値の渡し方をするべきでしょうか？

[LouiS0616]

追記のコードを試してみてください。

[LouiS0616]

ファイル名が正しく得られているか確認してから、imreadの可否を考えるべきかと。

[kaitokimura]

ファイル名は正しく得られていました。
openしてからimreadに渡す必要があるようですがまずカレントディレクトリに入れるとopencvがみつかりませんとなりそのためopenにわたすパスを絶対パスにした結果からはディレクトリですみたいな怒られ方をしました。

[LouiS0616]

imread(link3 + '/' + data)のように、
絶対パスで指定してもうまくいきませんか？

[LouiS0616]

ひごろC++でOpenCVを組んでいるので少し調べましたが、imreadより先にopenする必要はないようです。

[kaitokimura]

大変助かりました。絶対パスをjoinで作って見たのですがたぶんまちがっていたようです。Louiさんの教えてくださった通りに指定したら動きました。ただキーボードを２回ずつ押さなければ移動しないのですがこれは仕様でしょうか？実装の仕方が問題でしょうか？

[LouiS0616]

cv2.waitKey()を二回呼び出していませんか？

[kaitokimura]

key = cv2.waitKey()
if cv2.waitKey():
if key == 48:
shutil.move(link4,link0)
elif key == 49:
shutil.move(link4,link1)
elif key == 50:
shutil.move(link4,link2)
これしか関連記述はありません。

[LouiS0616]

二行目が不要です。

[kaitokimura]

ありがとうございました。不要だった部分を復習して見たいと思います。
効率が３倍にはなったので本当に助かりましたありがとうございました！！