track_eval

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, loc, "tracker", "data")
+        out_loc_dir = os.path.join(args.output, loc, args.tracker, "data")
         os.makedirs(out_loc_dir, exist_ok=True)
         metadata_path = os.path.join(args.metadata, loc + ".json")
         print(in_loc_dir)
@@ -101,7 +101,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)
 
     if config['use_associative']:
-        
+
         do_confidence_boost(inference, outputs, verbose=verbose)
 
         outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
@@ -142,6 +142,7 @@ def argument_parser():
     parser.add_argument("--detections", required=True, help="Path to frame directory. Required.")
     parser.add_argument("--output", required=True, help="Path to output directory. Required.")
     parser.add_argument("--metadata", required=True, help="Path to output directory. Required.")
+    parser.add_argument("--tracker", default='tracker', help="Path to output directory. Required.")
     return parser
 
 if __name__ == "__main__":

scripts/track_eval.py

@@ -1,6 +1,6 @@
 import project_path
 import argparse
-from infer_frames import main as infer
+from track_detection import main as track
 import sys
 sys.path.append('..')
 sys.path.append('../caltech-fish-counting')
@@ -15,8 +15,9 @@ def main(args):
     infer_args = Object()
     infer_args.metadata = "../frames/metadata"
     infer_args.frames = "../frames/images"
-    infer_args.output = "../frames/result"
+    infer_args.output = "../frames/result_testing"
     infer_args.weights = "models/v5m_896_300best.pt"
+    infer_args.tracker = 'tracker'
 
     config = {
         'conf_threshold': float(args.conf_threshold),
@@ -27,9 +28,9 @@ def main(args):
         'min_hits': int(args.min_hits)
     }
 
-    infer(infer_args, config=config, verbose=False)
+    track(infer_args, config=config, verbose=False)
 
-    evaluate("../frames/result_testing", "../frames/MOT", "../frames/metadata", "tracker", False)
+    evaluate(infer_args.output, "../frames/MOT", "../frames/metadata", infer_args.tracker, False)
 
 
 def argument_parser():