Skip to content

TotallyFine/SSD-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SSD Pytorch

SSD300的代码,论文地址Single Shot MultiBox Detector。代码主要是参考了pytorch-ssd,原版的代码不能直接运行在pytorch 0.4.0上,以及它还有一些其他的问题,我都改过了,并且加上了我的所有注释和理解,可以直接运行。

环境安装好之后,下载VOC2012数据集,用utils/process_voc.py处理VOC2012训练集和测试集的标注,生成voc12_test.txt voc_train.txt放在data/下。

然后下载预训练的VGG模型,使用utils/process_vgg.py来生成加载了部分VGG参数的SSD原始模型,并将模型放在model/下。

环境

  • cuda 8.0
  • cudnn 7.1.2
  • anaconda2
  • numpy 1.14.3
  • pytorch 0.4.0
  • torchvision 0.2.1
  • python 2.7

About

Single Shot MultiBox Detector in pytorch 0.4.0

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages