self.load_state_dict(torch.load(pretrained, map_location='cpu')['model'], strict=False)删除['model']后成功运行

参考2021/5/24 voc-test数据集 yolov4测试 - 平台梦 (pingtaimeng.com)

Logo

更多推荐