安装可视化
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 function with non-static forward method is deprecated. Please use new-style autograd function with static forward method.
参考:https://blog.csdn.net/baidu_39389949/article/details/107398256
修改如下:
detetion.py文件class Detect(Function)换成如下代码:
class Detect