#15 fix mot data

Merged
msstudy merged 1 commits from ArthurZhao/bytetrack:master into master 1 year ago
  1. +1
    -1
      yolox/data/mot.py

+ 1
- 1
yolox/data/mot.py View File

@@ -49,7 +49,7 @@ class MOTDataset:
self.data_dir = data_dir
self.json_file = json_file
self.input_dim = input_dim
self.coco = COCO(os.path.join(self.data_dir, "annotations", self.json_file))
self.coco = COCO(os.path.join(self.data_dir, "mix_mot_ch/annotations", self.json_file))
self.ids = self.coco.getImgIds()
self.class_ids = sorted(self.coco.getCatIds())
cats = self.coco.loadCats(self.coco.getCatIds())


Loading…
Cancel
Save