ssd_tensorflow_traffic_sign_detection

收录理由

对于想研究TensorFlow中如何搭建Single Shot MultiBox Detector的开发者，这个项目是一份很实在的参考。代码拆成几个直白的脚本，从数据准备一路走到训练和推理，目标也很聚焦：在图像和视频里找出并分类交通标志。作者不回避短板，明确说了目前只认停止和行人横道两种标志，训练用的是美国LISA交通标志数据集，而且模型存在过拟合，所以更适合当作学习素材或迁移学习的起点，而不是直接上生产的成品。项目里提到的在GTX 1080上跑到40-45帧每秒，也大致能让人估出优化NMS步骤后还有多少实时处理的余量。