おそらくlibgtk2がインストールされていない環境でcv2.waitKey() および cv2.destroyAllWindows() が動作しないので除外

2026-04-09 06:45:09 +00:00 · 2023-04-20 06:20:02 +09:00
parent ee5cec7530
commit e746829b5f
1 changed files with 2 additions and 2 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1445,8 +1445,8 @@ def debug_dataset(train_dataset, show_input_ids=False):
                    im = im[:, :, ::-1]  # RGB -> BGR (OpenCV)
                    if os.name == "nt":  # only windows
                        cv2.imshow("img", im)
-                    k = cv2.waitKey()
-                    cv2.destroyAllWindows()
+                        k = cv2.waitKey()
+                        cv2.destroyAllWindows()
                    if k == 27 or k == ord("s") or k == ord("e"):
                        break
            steps += 1