train.py

@@ -304,8 +304,8 @@ def run_train():
     config.logger.info("Network weights have been initialized...")
     network = YOLOLossCell(base_network, config)
     config.logger.info('Finish getting network...')
-    # config.data_root = os.path.join(config.data_dir)
-    ds = create_dataloader(config, image_dir=config.data_path, device_num=config.group_size, rank=config.rank)
+    config.data_root = os.path.join(config.data_dir, 'mix_mot_ch')
+    ds = create_dataloader(config, image_dir=config.data_root, device_num=config.group_size, rank=config.rank)
     # ds_test = get_val_dataset()
     config.logger.info('Finish loading training dataset! batch size:%s' % config.per_batch_size)
     config.steps_per_epoch = ds.get_dataset_size()

yolox/data/mot.py

@@ -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, "mix_mot_ch/annotations", self.json_file))
+		self.coco = COCO(os.path.join(self.data_dir, "annotations", self.json_file))
 		self.ids = self.coco.getImgIds()
 		self.class_ids = sorted(self.coco.getCatIds())
 		cats = self.coco.loadCats(self.coco.getCatIds())