From 0d76bee2ee158b46c1f9d0809c7f738812452be7 Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 7 Feb 2019 11:37:55 +0400 Subject: [PATCH] fix Trainer.py for debug --- mainscripts/Trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index f902c23..9a348f9 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -247,8 +247,9 @@ def previewThread (input_queue, output_queue): final = np.concatenate ( [final, lh_img], axis=0 ) final = np.concatenate ( [final, selected_preview_rgb], axis=0 ) + final = np.clip(final, 0, 1) - cv2.imshow ( 'Training preview', final) + cv2.imshow ( 'Training preview', (final*255).astype(np.uint8) ) is_showing = True if is_showing: