Spaces:
Runtime error
Runtime error
Commit
·
17e3572
1
Parent(s):
84f1118
AT testing
Browse files
lib/fish_eye/associative.py
CHANGED
@@ -225,9 +225,9 @@ class Associate(object):
|
|
225 |
for m in high_matched:
|
226 |
self.trackers[m[1]].update(high_dets[m[0], :])
|
227 |
|
228 |
-
if len(unmatched_trk_ids) > 0:
|
229 |
|
230 |
-
|
231 |
|
232 |
#for m in low_matched:
|
233 |
#self.trackers[m[1]].update(low_dets[m[0], :])
|
|
|
225 |
for m in high_matched:
|
226 |
self.trackers[m[1]].update(high_dets[m[0], :])
|
227 |
|
228 |
+
#if len(unmatched_trk_ids) > 0:
|
229 |
|
230 |
+
# low_matched, unmatched_low_dets, unmatched_trk_ids = associate_detections_to_trackers(low_dets, trks[unmatched_trk_ids], self.iou_threshold)
|
231 |
|
232 |
#for m in low_matched:
|
233 |
#self.trackers[m[1]].update(low_dets[m[0], :])
|
scripts/track_detection.py
CHANGED
@@ -101,7 +101,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
|
|
101 |
image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
|
102 |
|
103 |
|
104 |
-
if
|
105 |
|
106 |
low_outputs = do_suppression(inference, conf_thres=config['low_conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
107 |
low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
|
|
|
101 |
image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
|
102 |
|
103 |
|
104 |
+
if True:
|
105 |
|
106 |
low_outputs = do_suppression(inference, conf_thres=config['low_conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
107 |
low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
|