最近在学习目标检测算法,学到SSD系列 但是predictdou都没跑通,参考各种大佬的踩坑指南,终于运行成功了 ,这里默认该下载的都下载了,代码和训练好的权重文件,附上Bubbliiiing大佬的代码
**问题1:**报错Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd
查了下主要原因是pytorch版本问题,我不想降低版本,主要是我不会,我觉得很麻烦,哈哈哈,在我快要放弃的时候发现了一个大佬的博文,对这个问题讲的超级详细,可以直接参考,不过我在代码中没有改那么多,只是在nets\ssd.py文件中 detect后加.forword就可以了,如下图:
**问题2:**如何能用cpu运行起来,因为实验室就一个服务器,大家排队用,所以,emmmm排不上,为了能看效果只能用cpu跑了。
**解决方法:**目录下的ssd.py文件,cuda设成false且 map_location=‘cpu’
Pytorch搭建SSD遇到的问题(Bubbliiiing大佬的代码)
最新推荐文章于 2024-09-11 12:11:23 发布
博主分享了在学习SSD目标检测算法过程中遇到的PyTorch版本兼容问题解决方案,只需修改nets/ssd.py中detect函数为.forward,同时指导如何在CPU环境下运行。文章详述了问题定位和简易解决步骤。
摘要由CSDN通过智能技术生成