Update track_detection.py

scripts/track_detection.py

@@ -6,7 +6,7 @@ from datetime import datetime
 import torch
 import os
 from dataloader import create_dataloader_frames_only
-from inference import setup_model, do_detection, do_suppression, do_confidence_boost, format_predictions, do_tracking
+from inference import setup_model, do_detection, do_suppression, do_confidence_boost, format_predictions, do_tracking, do_associative_tracking
 from visualizer import generate_video_batches
 import json
 from tqdm import tqdm
@@ -102,6 +102,13 @@ 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)
 
+    print(len(outputs))
+    print(len(outputs[0]))
+    print(outputs[0][0])
+
+    results = do_associative_tracking(outputs, 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)
+
+
     if config['use_associative']:
 
         do_confidence_boost(inference, outputs, conf_power=config['boost_power'], conf_decay=config['boost_decay'], verbose=verbose)
@@ -110,10 +117,6 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
 
     all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height, verbose=verbose)
 
-    print(len(all_preds))
-    print(all_preds[0][0])
-
-    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)
 
     mot_rows = []
     for frame in results['frames']: