基于pytorch的SSD复现笔记

复现的代码来源于:

https://github.com/amdegroot/ssd.pytorch

复现的过程按照github的步骤进行:

但注意有几个坑:

1.首先是数据下载:

源码提供了脚本文件,下载数据,里面下载数据用的curl命令,但这个命令需要安装:

sudo apt install curl

否则报错还不好找错误,也可以使用wget命令下载。

两个命令的使用参考:https://blog.csdn.net/qq_33547191/article/details/88220199

2.cv2的安装:

详细参考我的另一篇博客:https://blog.csdn.net/qq_33547191/article/details/88223092

cv2的安装包名字不叫cv2,而是叫opencv-python

3.pytorch版本需要为0.4.1,python版本为python3.6(千万别用3.70)

python3.6的anaconda安装,见我的一篇博客:

https://blog.csdn.net/qq_33547191/article/details/88223092

4.loss计算,需要加一句:

在multibox_loss.py的97行前加:

   loss_c = loss_c.view(num, -1)#The line added
   loss_c[pos] = 0  # filter out pos boxes for now

5.加上之后

N会报数据类型有问题,

 N = num_pos.data.sum().float()

做完这几步就可以训练了。

这个程序中eval和test都没有将检测结果框可视化,但在demo文件夹中提供了demo的代码,在使用这部分代码的时候有一个BUG,总是报内存不足。

解决办法:将python3.6切换到python2.7

结果:

 

从结果上可以看到对于多人检测还是有两个很明显的人没有检测出来。

 

  • 4
    点赞
  • 32
    收藏
    觉得还不错? 一键收藏
  • 28
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值