ssd_keras

收录理由

这个Keras版本的SSD目标检测实现，出发点很明确：把模型结构讲清楚，而不是当作黑盒来用。项目把重心放在低层理解上，所以代码注释和教程写得很详细，除了完整的SSD300和SSD512，还附带一个精简的SSD7小模型，可以在中端GPU上从头训练，适合处理不太复杂的检测任务或做快速测试。从原始Caffe实现移植过来的权重能复现相同的mAP，意味着你在这边训练出来的模型，精度上应该和原版一致。微调用的Jupyter笔记本可以帮助你从预训练权重里抽取与自定义数据集兼容的子集，基础网络模板也允许替换成自己的结构。不过要留意，代码基于较老的Keras API，而且多年没更新了，把它当作研究或教学用的参考文档更合适，不太适合直接拿去做生产系统的基础。