oskarastrom commited on
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
- 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], :])
 
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 False:
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)