tiny changes

.gitignore

@@ -6,3 +6,5 @@
 logs/
 demo/data/
 labels/
+*.mp4
+*.json

re3_utils/util/drawing.py

@@ -33,7 +33,7 @@ def subplot(plots, rows, cols, outputWidth, outputHeight, border=BORDER,
             if im.dtype != np.uint8 or len(im.shape) < 3:
                 im = im.astype(np.float32)
                 im -= np.min(im)
-                im *= 255 / np.max(im)
+                im *= 255 / max(np.max(im), 0.0001)
                 im = 255 - im.astype(np.uint8)
             if len(im.shape) < 3:
                 im = cv2.applyColorMap(
@@ -102,8 +102,8 @@ def subplot(plots, rows, cols, outputWidth, outputHeight, border=BORDER,
                     font=FANCY_FONT)
             im = np.array(im)
         else:
-            cv2.putText(im, titles[0], (10, 30), CV_FONT, 128, [0,0,0], 4)
-            cv2.putText(im, titles[0], (10, 30), CV_FONT, 128, [1,1,1], 1)
+            cv2.putText(im, titles[0], (10, 30), CV_FONT, .5, [0,0,0], 4)
+            cv2.putText(im, titles[0], (10, 30), CV_FONT, .5, [255,255,255], 1)
 
     return im
 

training/test_net.py

@@ -20,7 +20,6 @@ from tracker import re3_tracker
 from constants import CROP_PAD
 from constants import CROP_SIZE
 from constants import MAX_TRACK_LENGTH
-from constants import OUTPUT_SIZE
 
 # Display constants
 from constants import OUTPUT_WIDTH