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等。您可以用自己的代码