プログラムの中断ができない

Question

python初心者です。
画像ファイルを目視し、key操作(1or2or3)でそれぞれclass1,class2,class3に振り分けるプログラムを考えています。

1or2or3以外のkey操作が行われたらそのまま画像表示させ、1or2or3が入力されたら次の画像を表示させたいと思っております。
フォルダの中のファイルがなくなったら、プログラムは終了しますが、途中で中断させようとしたときにCtrl+cを受け付けません。
途中でプログラムを終了させるためにCtrl+zで強制終了しなくてはいけなく、Ctrl+cでプログラムを中断させたいのです。
while Trueが悪さをしているように思いますが、elseのあとにbreakをしてしまうと1or2or3以外のキー操作がされたときでも画像が次のものになってしまいます。

まとめますと
・1or2or3以外のキー操作をされたら、画像表示を維持したい。
・プログラムの途中でCtrl +c を受け付け正常処理させたい。

import cv2
import os
import glob
import shutil

sort1='/home/user/Desktop/code.test/class1'
sort2='/home/user/Desktop/code.test/class2'
sort3='/home/user/Desktop/code.test/class3'
target_dir='/home/user/Desktop/code.test'

#保存先ディレクトリがなかったら作成する
if not os.path.exists(sort1):
    os.mkdir(sort1)
if not os.path.exists(sort2):
    os.mkdir(sort2)
if not os.path.exists(sort3):
    os.mkdir(sort3)


filenames = os.listdir(target_dir)
for file in filenames:
    multiple = 2
    img = cv2.imread(target_dir+'/'+file)
    target_file=target_dir+'/'+file
 
    if img is None:
       print('through')
    else:
       height = img.shape[0]
       width = img.shape[1]
       img = cv2.resize(img , (int(width * multiple), int(height * multiple)))
       cv2.imshow("image", img)
       while True: 
         key = cv2.waitKey()      
         if key == 49:
            shutil.move(target_file,sort1)
            break
         elif key == 50:
            shutil.move(target_file,sort2)
            break
         elif key == 51:
            shutil.move(target_file,sort3)
            break
     #他のキーが入力されたらそのままにする
         else:
            print('pass', str(key))

Comments

[8524ba23]

どのような操作（キー入力）によってどのような動作をさせたいのかが明確でないように思います。
「～したら～したい」を具体的に記載ください。

Answer 1

「プログラムの途中でCtrl +c を受け付け正常処理させたい」

「プログラムを正常終了」なのか「何もせず次の画像へ移行」なのかがいまいち不明ですが、基本的な処理の流れのコードを提示します。
このコードでは

• is_exitフラグを用いて途中で終了できるように
• waitKeyは入力待ちする（無駄にループを回さない）ように

しています。その結果

キー	処理
1,2,3	振り分けて次の画像へ
n	何もせず次の画像へ
e	何もせずプログラム終了
それ以外	再入力（今の画像表示を維持）

という動作をします。

import cv2
import numpy as np

is_exit = False
for i in range(10):
    if is_exit:
        break

    print(f'i={i}')
    img = np.ones((64,48,3), np.uint8) * i * 25
    cv2.imshow("image", img)

    while True:
        # OpenCV Python: How to detect if a window is closed?
        # https://stackoverflow.com/questions/35003476/opencv-python-how-to-detect-if-a-window-is-closed
        if cv2.getWindowProperty('image', cv2.WND_PROP_VISIBLE) < 1:
            print('ユーザーがウインドウを閉じたので終了')
            is_exit = True
            break

        key = cv2.waitKey(1000) # 1秒待つ

        if key in [49,50,51]:
            print(f'{chr(key)}の処理をして次の画像へ')
            break
        elif key == ord('n'):
            print('何もせず次の画像へ')
            break
        elif key == ord('e'):
            print('何もせず終了')
            is_exit = True
            break
        else:
            print('未入力or無効な入力なので再入力')

cv2.destroyAllWindows()

Comments

[python_begin22]

ご連絡ありがとうございます。
こちらのコードを試させていただきました。
「e」が入力されたら、フラクで終了するといったこと参考にさせていただきます。

こちらのコードでも同様ですが、実行中にCtrl + cを受け付けないのですが、こちらの原因をご教示いただければ幸いです。
初心者で申し訳ございません。

[8524ba23]

画像ウインドウを表示しているので、GUI上でのキー処理が優先実行されるようです。
当方Windows環境では「Ctrl+C」にて画像がクリップボードにコピーされます。
なお、回答コードにてユーザーが画像ウインドウを×ボタンなどで閉じた場合にキー入力ループが抜けられなくなっていたので修正しました。

[python_begin22]

回答ありがとうございます。

私はubuntu環境ですが、画像ウインドウを☓ボタンで閉じても終了処理されず操作を受け付けない状態になりました。
Windows環境と実行結果が異なるのでしょうか？

[8524ba23]

OpenCVがバックエンドで使っているGUIはOSなど環境により異なるので、その結果、挙動が異なっていると思います。

[python_begin22]

承知いたしました。

ありがとうございます。

[8524ba23]

ウインドウが閉じられているかを明示的にチェックするように回答コードを修正しました。
ubuntu環境での動作は未確認ですが、ユーザーがウインドウ閉じた後、遅くとも１秒後にはプログラム終了すると思います。

[python_begin22]

ubnutuでも試して見ましたが完璧に動作しました。
大変助かりました。
canさんのようにコードすらすらかけるようになりたいです。