fix Trainer.py for debug

2019-02-07 11:37:55 +04:00
parent cb96c2a51a
commit 0d76bee2ee
1 changed files with 2 additions and 1 deletions
--- 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: