Spaces:
Runtime error
Runtime error
oskarastrom
commited on
Commit
•
53af29c
1
Parent(s):
b59bd4e
Removed prints
Browse files- inference.py +0 -11
- lib/fish_eye/tracker.py +0 -1
inference.py
CHANGED
@@ -203,11 +203,6 @@ def format_predictions(image_shapes, outputs, width, height, gp=None, batch_size
|
|
203 |
gp: a callback function which takes as input 1 parameter, (int) percent complete
|
204 |
prep_for_marking: re-index fish for manual marking output
|
205 |
"""
|
206 |
-
|
207 |
-
print(type(image_shapes))
|
208 |
-
print(len(image_shapes))
|
209 |
-
print(type(outputs))
|
210 |
-
print(len(outputs))
|
211 |
|
212 |
if (gp): gp(0, "Formatting...")
|
213 |
# keep predictions to feed them ordered into the Tracker
|
@@ -219,12 +214,6 @@ def format_predictions(image_shapes, outputs, width, height, gp=None, batch_size
|
|
219 |
if gp: gp(batch_i / len(image_shapes), pbar.__str__())
|
220 |
|
221 |
batch_shapes = image_shapes[batch_i]
|
222 |
-
print(type(batch_shapes))
|
223 |
-
print(len(batch_shapes))
|
224 |
-
print(type(batch))
|
225 |
-
print(len(batch))
|
226 |
-
print(batch_shapes[0])
|
227 |
-
print(batch[0])
|
228 |
|
229 |
# Format results
|
230 |
for si, pred in enumerate(batch):
|
|
|
203 |
gp: a callback function which takes as input 1 parameter, (int) percent complete
|
204 |
prep_for_marking: re-index fish for manual marking output
|
205 |
"""
|
|
|
|
|
|
|
|
|
|
|
206 |
|
207 |
if (gp): gp(0, "Formatting...")
|
208 |
# keep predictions to feed them ordered into the Tracker
|
|
|
214 |
if gp: gp(batch_i / len(image_shapes), pbar.__str__())
|
215 |
|
216 |
batch_shapes = image_shapes[batch_i]
|
|
|
|
|
|
|
|
|
|
|
|
|
217 |
|
218 |
# Format results
|
219 |
for si, pred in enumerate(batch):
|
lib/fish_eye/tracker.py
CHANGED
@@ -124,7 +124,6 @@ class Tracker:
|
|
124 |
|
125 |
fish_entry['travel_dist'] = Tracker.get_travel_distance(start_bbox, end_bbox, json_data['image_meter_width'], json_data['image_meter_height'])
|
126 |
|
127 |
-
print(boxes[0])
|
128 |
fish_entry['start_frame_index'] = boxes[0][1]
|
129 |
fish_entry['end_frame_index'] = boxes[-1][1]
|
130 |
fish_entry['color'] = Tracker.selectColor(track_id)
|
|
|
124 |
|
125 |
fish_entry['travel_dist'] = Tracker.get_travel_distance(start_bbox, end_bbox, json_data['image_meter_width'], json_data['image_meter_height'])
|
126 |
|
|
|
127 |
fish_entry['start_frame_index'] = boxes[0][1]
|
128 |
fish_entry['end_frame_index'] = boxes[-1][1]
|
129 |
fish_entry['color'] = Tracker.selectColor(track_id)
|