mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

set required=True for dataset arguments

pull/119/head
KaiyangZhou 2018-11-05 21:25:09 +00:00
parent 02bac86c71
commit 9726b8bb96
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchreid/data_manager.py
View File

@ -14,7 +14,7 @@ class ImageDataManager(object):
workers, pin_memory, **kwargs):
self.train_names = train_names
self.test_names = test_names if test_names is not None else train_names
self.test_names = test_names
self.train = []
self.num_train_pids = 0

4
train_imgreid_xent.py
View File

@ -31,8 +31,8 @@ parser = argparse.ArgumentParser(description='Train image model with cross entro
# Datasets
parser.add_argument('--root', type=str, default='data',
help="root path to data directory")
parser.add_argument('-s', '--source', type=str, default=['market1501'], nargs='+')
parser.add_argument('-t', '--target', type=str, nargs='+')
parser.add_argument('-s', '--source', type=str, required=True, nargs='+')
parser.add_argument('-t', '--target', type=str, required=True, nargs='+')
parser.add_argument('-j', '--workers', default=4, type=int,
help="number of data loading workers (default: 4)")
parser.add_argument('--height', type=int, default=256,