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

fix bug about changing cfg values

pull/224/head
liaoxingyu 2020-08-10 10:29:24 +08:00
parent d516d59fb5
commit 57ebf75f83
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fastreid/data/build.py
View File

@ -19,6 +19,9 @@ _root = os.getenv("FASTREID_DATASETS", "datasets")
def build_reid_train_loader(cfg):
cfg = cfg.clone()
cfg.defrost()
train_items = list()
for d in cfg.DATASETS.NAMES:
dataset = DATASET_REGISTRY.get(d)(root=_root, combineall=cfg.DATASETS.COMBINEALL)
@ -26,7 +29,6 @@ def build_reid_train_loader(cfg):
dataset.show_train()
train_items.extend(dataset.train)
cfg.defrost()
iters_per_epoch = len(train_items) // cfg.SOLVER.IMS_PER_BATCH
cfg.SOLVER.MAX_ITER *= iters_per_epoch
train_transforms = build_transforms(cfg, is_train=True)
@ -57,6 +59,9 @@ def build_reid_train_loader(cfg):
def build_reid_test_loader(cfg, dataset_name):
cfg = cfg.clone()
cfg.defrost()
dataset = DATASET_REGISTRY.get(dataset_name)(root=_root)
if comm.is_main_process():
dataset.show_test()