Skip to content

Commit b625a4d

Browse files
authored
Update track_yolov5.py
1 parent 83633f2 commit b625a4d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tracker/track_yolov5.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ def main(opts, cfgs):
210210

211211
# Run code
212212
evaluator = trackeval.Evaluator(eval_config)
213-
dataset_list = [trackeval.datasets.MotChallenge2DBox(dataset_config)] if opts.datasets in ['mot', 'uavdt'] else [trackeval.datasets.VisDrone2DBox(dataset_config)]
213+
dataset_list = [trackeval.datasets.MotChallenge2DBox(dataset_config)] if opts.dataset in ['mot', 'uavdt'] else [trackeval.datasets.VisDrone2DBox(dataset_config)]
214214
metrics_list = []
215215
for metric in [trackeval.metrics.HOTA, trackeval.metrics.CLEAR, trackeval.metrics.Identity, trackeval.metrics.VACE]:
216216
if metric.get_name() in metrics_config['METRICS']:

0 commit comments

Comments
 (0)