数据集:ImageNet2012 validation
例:/public/opendas/DL_DATA/ImageNet-pytorch/val
使用resnet50开源模型,代码运行中自动下载:https://download.pytorch.org/models/resnet50-0676ba61.pth
单卡即可推理
测试方法:
export HIP_VISIBLE_DEVICES=0 # 指定卡号，不指定时默认使用第一张卡
python infer_resnet50.py
