打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
仿照yolov5官方示例移植yolov7,在pytorch_yolov7_inference.py一开始调用加载原pt的load_dict函数后报错,提示找不到GPU设备,可以添加map_location=‘cpu’。添加后能正常进行推理,但为什么和寒武纪仓库提供的pytorch_yolov5_inference.py写的一样,运行yolov5是正常的,但yolov7就报错。
热门帖子
精华帖子