From 4638af3fb5dcf8478af5995d3be40483e7c050bd Mon Sep 17 00:00:00 2001
From: Daniel Gordon <xkcd@cs.washington.edu>
Date: Tue, 28 Nov 2017 17:44:17 -0800
Subject: [PATCH] tiny changes
---
.gitignore | 2 ++
re3_utils/util/drawing.py | 6 +++---
training/test_net.py | 1 -
3 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/.gitignore b/.gitignore
index 38d7ebb..b1d78f9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,3 +6,5 @@
logs/
demo/data/
labels/
+*.mp4
+*.json
diff --git a/re3_utils/util/drawing.py b/re3_utils/util/drawing.py
index 929bb2a..d123a69 100644
--- a/re3_utils/util/drawing.py
+++ b/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
diff --git a/training/test_net.py b/training/test_net.py
index 9ae2e41..78d9a87 100644
--- a/training/test_net.py
+++ b/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
--
GitLab