Fixed tracking

inference.py

@@ -323,7 +323,6 @@ def do_tracking(all_preds, image_meter_width, image_meter_height, gp=None, max_a
             pbar.update(1)
 
     json_data = tracker.finalize(min_length=min_length)
-    print(json_data['fish'])
 
     return json_data
 

scripts/detect_frames.py

@@ -73,7 +73,6 @@ def detect(in_seq_dir, out_seq_dir, model, device, verbose):
         'width': width,
         'height': height
     }
-    print('created object')
 
     with open(os.path.join(out_seq_dir, 'pred.json'), 'w') as f:
         json.dump(json_obj, f)

scripts/track_detection.py

@@ -45,7 +45,7 @@ def main(args, config={}, verbose=True):
     for loc in locations:
 
         in_loc_dir = os.path.join(args.detections, loc)
-        out_loc_dir = os.path.join(args.output, "tracker", "data", loc)
+        out_loc_dir = os.path.join(args.output, loc, "tracker", "data")
         os.makedirs(out_loc_dir, exist_ok=True)
         metadata_path = os.path.join(args.metadata, loc + ".json")
         print(in_loc_dir)
@@ -81,10 +81,6 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
     device = select_device(device_name, batch_size=32)
     inference = torch.load(inference_path, map_location=device)
 
-    print(type(inference))
-    print(len(inference))
-    print(type(inference[0]))
-
     # read detection
     with open(json_path, 'r') as f:
         detection = json.load(f)
@@ -109,7 +105,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
 
         outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 
-    all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height)
+    all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height, verbose=verbose)
 
     results = do_tracking(all_preds, image_meter_width, image_meter_height, min_length=config['min_length'], max_age=config['max_age'], iou_thres=config['iou_threshold'], min_hits=config['min_hits'], verbose=verbose)