From cf8315c5b3f057badcfbb193d6ff403d1678959c Mon Sep 17 00:00:00 2001 From: "zuchen.wang" Date: Wed, 17 Nov 2021 16:40:25 +0800 Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E4=B8=BA1102=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- projects/FastShoe/fastshoe/trainer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/projects/FastShoe/fastshoe/trainer.py b/projects/FastShoe/fastshoe/trainer.py index 16e86fd..82ac0e0 100644 --- a/projects/FastShoe/fastshoe/trainer.py +++ b/projects/FastShoe/fastshoe/trainer.py @@ -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))