mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

改为1102数据

pull/608/head
zuchen.wang 2021-11-17 16:40:25 +08:00
parent 339b5f2188
commit cf8315c5b3
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
projects/FastShoe/fastshoe/trainer.py
View File

@ -28,7 +28,8 @@ class PairTrainer(DefaultTrainer):
transforms = build_transforms(cfg, is_train=True)
img_root=os.path.join(_root, 'shoe_crop_all_images')
anno_path=os.path.join(_root, 'labels/1019/1019_clean_train.json')
anno_path=os.path.join(_root, 'labels/1102/train_1102.json')
cls._logger.info('Loading {} with {}.'.format(img_root, anno_path))
datasets = []
for d in cfg.DATASETS.NAMES:
@ -48,8 +49,8 @@ class PairTrainer(DefaultTrainer):
transforms = build_transforms(cfg, is_train=False)
if dataset_name == 'PairDataset':
img_root = os.path.join(_root, 'shoe_crop_all_images')
val_json = os.path.join(_root, 'labels/1019/1019_clean_val.json')
test_json = os.path.join(_root, 'labels/1019/1019_clean_test.json')
val_json = os.path.join(_root, 'labels/1102/validation_1102.json')
test_json = os.path.join(_root, 'labels/1102/test_1102.json')
anno_path, mode = (test_json, 'test') if cfg.eval_only else (val_json, 'val')
cls._logger.info('Loading {} with {} for {}.'.format(img_root, anno_path, mode))