ssd_pytorch

by acm5656

acm5656 / ssd_pytorch

ssd implented by pytorch

132 Stars 23 Forks Last release: Not found 29 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

ssd_pytorch

首先说下实现原因,大佬写的代码太难读了,对于新手太不友好,而且还是英文的,让我这英语渣渣看起来头疼,为了学习和帮助其他像我一样的人,复现了一下代码,并简化部分功能

首先发下代码的参考连接 ssd-pytorch
论文地址就不发了,可以自行百度,我的博客ssd解读地址在这里,可以进行参考学习
代码解读地址在这里

运行环境:

1.python 3.6
2.pytorch 0.4.1
3.python-opencv

说明

预训练的权重文件vgg_16 提取码:rdbn
具体的配置文件请看Config.py文件
训练运行python3.6 Train.py
已经训练好的模型ssd_300
测试运行python3.6 Test.py

运行结果

demo

注:

1.目前的版本是作者学习ssd模型并参考他人代码尝试复现的结果,所以代码鲁棒性较低,望见谅
2.目前版本只兼容voc的数据集
3.如果帮到你啦,麻烦给个star,谢谢啦
4.目前作者已经转行做其他方向了,这个也不再维护了,但是可以确保在环境一致的情况下是可以运行的

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.