文件名称：SSD-pytorch:pytorch 0.4.0中的单发MultiBox检测器
文件大小：19KB
文件格式：ZIP
更新时间：2024-05-18 18:10:13
Python
SSD Pytorch SSD300的代码，论文地址。代码主要是参考了，原版的代码不能直接运行在pytorch 0.4.0上，以及它还有一些其他的问题，我都改过了，并且加上了我的所有注释和理解，可以直接运行。 环境安装好之后，下载VOC2012数据集，用utils/process_voc.py处理VOC2012训练集和测试集的标注，生成voc12_test.txt voc_train.txt放在data/下。 然后下载预训练的，使用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
【文件预览】：
SSD-pytorch-master
----data()
--------dataset.py(6KB)
--------Box.py(11KB)
--------__init__.py(71B)
----model()
--------MultiBoxLoss.py(8KB)
--------SSD.py(7KB)
--------MultiBox.py(2KB)
--------__init__.py(111B)
----test.py(1016B)
----train.py(4KB)
----config.py(2KB)
----README.md(875B)
----utils()
--------process_voc.py(1KB)
--------process_vgg.py(853B)