Merge pull request kohya-ss#429 from tsukimiya/hotfix/debug_dataset_l…

…inux_support Refixed --debug_dataset option to work in non-Windows environments
DarksealStudios · Apr 21, 2023 · 9f8f27f · 9f8f27f
2 parents 589a90b + e746829
commit 9f8f27f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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