Spaces:
Runtime error
Runtime error
Commit
·
73afcbb
1
Parent(s):
66f8a6f
updated hyperparams
Browse files
scripts/{inferEval.py → infer_eval.py}
RENAMED
@@ -29,7 +29,7 @@ def main(args):
|
|
29 |
|
30 |
infer(infer_args, config=config, verbose=False)
|
31 |
|
32 |
-
evaluate("../frames/
|
33 |
|
34 |
|
35 |
def argument_parser():
|
|
|
29 |
|
30 |
infer(infer_args, config=config, verbose=False)
|
31 |
|
32 |
+
evaluate("../frames/result", "../frames/MOT", "../frames/metadata", "tracker", False)
|
33 |
|
34 |
|
35 |
def argument_parser():
|
scripts/track_detection.py
CHANGED
@@ -28,8 +28,8 @@ def main(args, config={}, verbose=True):
|
|
28 |
"""
|
29 |
|
30 |
# setup config
|
31 |
-
if "conf_threshold" not in config: config['conf_threshold'] = 0.001
|
32 |
-
if "nms_iou" not in config: config['nms_iou'] = 0.6
|
33 |
if "min_length" not in config: config['min_length'] = 0.3
|
34 |
if "max_age" not in config: config['max_age'] = 20
|
35 |
if "iou_threshold" not in config: config['iou_threshold'] = 0.01
|
@@ -101,6 +101,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
|
|
101 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
102 |
|
103 |
if config['use_associative']:
|
|
|
104 |
do_confidence_boost(inference, outputs, verbose=verbose)
|
105 |
|
106 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
|
|
28 |
"""
|
29 |
|
30 |
# setup config
|
31 |
+
if "conf_threshold" not in config: config['conf_threshold'] = 0.3#0.001
|
32 |
+
if "nms_iou" not in config: config['nms_iou'] = 0.3#0.6
|
33 |
if "min_length" not in config: config['min_length'] = 0.3
|
34 |
if "max_age" not in config: config['max_age'] = 20
|
35 |
if "iou_threshold" not in config: config['iou_threshold'] = 0.01
|
|
|
101 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
102 |
|
103 |
if config['use_associative']:
|
104 |
+
|
105 |
do_confidence_boost(inference, outputs, verbose=verbose)
|
106 |
|
107 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|