From 949ee6fcc9b1eae195f7ae5191e720c5eb8e0e5d Mon Sep 17 00:00:00 2001
From: tsukimiya <71832+tsukimiya@users.noreply.github.com>
Date: Fri, 3 Feb 2023 00:37:27 +0900
Subject: [PATCH] Fixed --debug_dataset option to work in non-Windows
 environments

---
 library/train_util.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/library/train_util.py b/library/train_util.py
index 85b58d7e..cd3d0624 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -747,7 +747,8 @@ def debug_dataset(train_dataset, show_input_ids=False):
         im = ((im.numpy() + 1.0) * 127.5).astype(np.uint8)
         im = np.transpose(im, (1, 2, 0))                # c,H,W -> H,W,c
         im = im[:, :, ::-1]                             # RGB -> BGR (OpenCV)
-        cv2.imshow("img", im)
+        if os.name == 'nt':                             # only windows
+          cv2.imshow("img", im)
         k = cv2.waitKey()
         cv2.destroyAllWindows()
         if k == 27: