def main():
args = parse_args()
cfg = Config.fromfile(args.config)
if args.work_dir is not None:
cfg.work_dir = args.work_dir
if args.checkpoint is not None:
cfg.load_from = args.checkpoint
# init distributed env first
if args.launcher == 'none':
distributed = False
else:
distributed = True
init_dist(args.launcher, **cfg.dist_params)
# init logger
logger = get_root_logger(cfg.log_level)
logger.info('Distributed test: {}'.format(distributed))
# build model and load checkpoint
if args.stage == 'GMM':
# test geometric matching
# data loader
dataset = get_dataset(cfg.data.test.GMM)
print('GMM dataset loaded')
model = build_geometric_matching(cfg.GMM)
print('GMM model built')
load_checkpoint(model, cfg.load_from, map_location='cpu')
print('load checkpoint from: {}'.format(cfg.load_from))
test_geometric_matching(model,
dataset,
cfg,
distributed=distributed,
validate=False,
logger=logger)
elif args.stage == 'TOM':
# test tryon module
dataset = get_dataset(cfg.data.test.TOM)
print('TOM dataset loaded')
model = build_tryon(cfg.TOM)
print('TOM model built')
load_checkpoint(model, cfg.load_from, map_location='cpu')
print('load checkpoint from: {}'.format(cfg.load_from))
test_tryon(model,
dataset,
cfg,
distributed=distributed,
validate=False,
logger=logger)
def main():
args = parse_args()
cfg = Config.fromfile(args.config)
if args.work_dir is not None:
cfg.work_dir = args.work_dir
if args.checkpoint is not None:
cfg.load_from = args.checkpoint
# init distributed env first
if args.launcher == 'none':
distributed = False
else:
distributed = True
init_dist(args.launcher, **cfg.dist_params)
# init logger
logger = get_root_logger(cfg.log_level)
logger.info('Distributed test: {}'.format(distributed))
# data loader
dataset = get_dataset(cfg.data.test)
print('dataset loaded')
# build model and load checkpoint
model = build_predictor(cfg.model)
print('model built')
load_checkpoint(model, cfg.load_from, map_location='cpu')
print('load checkpoint from: {}'.format(cfg.load_from))
# test
test_predictor(
model,
dataset,
cfg,
distributed=distributed,
validate=args.validate,
logger=logger)