args = parser.parse_args()
return args
if __name__ == '__main__':
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
args = parse_args()
cfg = mmcv.Config.fromfile(args.config)
model = build_recognizer(
cfg.model, train_cfg=None, test_cfg=cfg.test_cfg)
# model.to('cuda:0')
# model.eval()
load_checkpoint(model, args.checkpoint, strict=True)
model = MMDataParallel(model, device_ids=[0])
model.cfg = cfg
model.eval()
"""
图片识别
"""
# from mmcv.runner import load_checkpoint, parallel_test, obj_from_dict
# from mmaction import datasets
# from mmaction.datasets import build_dataloader
# dataset = obj_from_dict(cfg.data.test, datasets, dict(test_mode=True))
# data_loader = build_dataloader(
# dataset,
# imgs_per_gpu=1,
# workers_per_gpu=cfg.data.workers_per_gpu,
# num_gpus=1,
# dist=False,
# shuffle=False)
