SSD:PyTorch中高质量快速模块化的SSD参考实现 源码
PyTorch 1.0中高质量,快速,模块化的SSD参考实现 该存储库实现 。 实施受项目 , 和影响很大。 该存储库旨在成为基于SSD的研究的代码库。 示例SSD输出(vgg_ssd300_voc0712)。 损失 学习率 指标 强调 PyTorch 1.0 :支持PyTorch 1.0或更高版本。 多GPU训练和推论:我们使用DistributedDataParallel ,您可以使用任意GPU进行训练或测试,训练方案将相应更改。 模块化:轻松添加自己的模块。 我们抽象了backbone , Detector , BoxHead , BoxPredictor等。您可以用自己的代码
文件列表
SSD-master.zip
(预估有个85文件)
SSD-master
MANIFEST.in
33B
figures
losses.png
39KB
lr.png
13KB
metrics.png
16KB
004545.jpg
53KB
train.py
4KB
configs
vgg_ssd512_voc0712.yaml
683B
暂无评论