ssd.pytorch使用

参考:ssd.pytorch源码

安装可视化

pip install visdom

pip安装的时候用镜像源不可以安装,很奇怪。我把镜像源去掉了安装成功。
然后输入命令python -m visdom.server开启服务端程序。在web上输入地址10.1.130.211:8097,即可在web看到可视化过程。
在这里插入图片描述

源码运行

下载数据集

数据集存放位置~/data/
在这里插入图片描述
VOC数据集下载下来解压就已经是VOC格式,可以直接跑起来。

源码修改

细节参考:(5条消息) ssds目标检测项目中遇到的一些bug_羊藤枝的博客-CSDN博客他讲的比较详细。

由于源码的pytorch环境还是我装的新,所以代码上需要一
些修改。

训练时

问题一:
在这里插入图片描述去train.py中,
改成如下:
在这里插入图片描述
问题二:
在这里插入图片描述
去train.py中,改成如下:
在这里插入图片描述
问题三:
在这里插入图片描述
去ssd.py中,改成如下:
在这里插入图片描述
去train.py中,改成如下:
在这里插入图片描述
问题四:
在这里插入图片描述
去multibox_loss.py,改成如下:
在这里插入图片描述

在这里插入图片描述
问题五:
train.py代码中,在迭代过程中,每次执行batch张图片,通过images, targets = next(batch_iterator)读取图片时,如果next()中没有数据后会触发Stoplteration异常,使用下面语句替换 images, targets = next(batch_iterator)将解决这种异常问题。
在这里插入图片描述
去train.py,改成如下
在这里插入图片描述

测试时

问题一:
RuntimeError: Legacy autograd func

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值