SSD-Tensorflow

收录理由

这份代码库把SSD论文里的方法用TensorFlow重新实现了一遍，对于想弄清楚现代目标检测器内部机制的人来说，是很值得反复阅读的参考实现。它基于VGG骨干网络复现了Single Shot MultiBox Detector的原始结构，支持300和512两种输入尺寸，你可以沿着代码一步步看到单个网络如何在多尺度特征图上产生预测框，再通过非极大值抑制完成筛选。项目把数据集处理、网络定义和预处理拆成了不同模块，还附带了从Caffe转换来的预训练权重以及一个完整的notebook示例，能直接跑通从加载图片到输出检测结果的整个流程。这种组织方式对研究算法原理、在Pascal VOC上做微调，或者尝试替换其他骨干网络都挺有帮助。不过要坦诚地说，这是一个研究时期的TensorFlow 1.x项目，大概在2021年后就没有再积极维护了，所以更适合用来学习和参考，而不是直接部署到生产环境。