Skip to content

Commit 064cf7d

Browse files
authored
Update track_yolov5.py
1 parent ae2ae84 commit 064cf7d

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

tracker/track_yolov5.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -35,22 +35,6 @@
3535
import tracker_dataloader
3636
import trackeval
3737

38-
"""
39-
DATASET_ROOT = '/data/wujiapeng/datasets/VisDrone2019/VisDrone2019' # your dataset root
40-
# DATASET_ROOT = '/data/wujiapeng/datasets/'
41-
42-
# CATEGORY_NAMES = ['car']
43-
CATEGORY_NAMES = ['car', 'van', 'truck', 'bus']
44-
# CATEGORY_NAMES = ['pedestrain', 'people', 'bicycle', 'car', 'van', 'truck', 'tricycle', 'awning-tricycle', 'bus', 'motor']
45-
CATEGORY_DICT = {i: CATEGORY_NAMES[i] for i in range(len(CATEGORY_NAMES))} # show class
46-
47-
# IGNORE_SEQS = []
48-
IGNORE_SEQS = ['uav0000073_00600_v', 'uav0000088_00290_v', 'uav0000073_04464_v'] # ignore seqs
49-
50-
# NOTE: ONLY for yolo v5 model loader(func DetectMultiBackend)
51-
YAML_DICT = {'visdrone': './data/Visdrone_car.yaml',
52-
'uavdt': './data/UAVDT.yaml'}
53-
"""
5438

5539
def set_basic_params(cfgs):
5640
global CATEGORY_DICT, DATASET_ROOT, CERTAIN_SEQS, IGNORE_SEQS, YAML_DICT

0 commit comments

Comments
 (0)