keras-yolo2

收录理由

训练自定义目标检测器，多数人会被框架配置折腾得头疼，而这个仓库直接给出了一套能跑的 YOLOv2 实现，基于 Keras 和 TensorFlow 后端，拿到自己的 VOC 格式标注和图片文件夹就能开始训练。骨干网络可以在 MobileNet、SqueezeNet、InceptionV3、ResNet50 之间切换，硬件资源有限时，牺牲一点精度换取更快的推理速度，很实用。仓库里附带浣熊、袋鼠、手部、红细胞检测等示例，相当于现成的起步模板；还有一个完全在浏览器里运行的浣熊检测演示，直观展示模型效果。对于想搞懂训练流程各部分如何衔接、而不只是拿现成黑箱工具的人来说，数据准备和训练步骤的讲解提供了很实在的入门路径。