目标检测训练框架
研究lufficc/SSD源码时,发现其实现了一个目标检测训练框架,进行深入研究后整理成一个新的训练框架
框架概述
将目标检测训练框架分为以下内容:
- 配置模块
- 日志模块
- 数据模块
- 检测器模块
- 优化器模块
- 辅助模块
该训练框架能够提供以下功能:
- 功能模块化:适用于不同算法的替换和移植
- 配置规范化:统一配置训练参数
- 日志读写:详细打印训练和测试结果,包含了文本日志和可视化日志
- 可持续训练:通过定期检查点保存,实现中断训练后的恢复训练
框架实现
模块实现地址:object-detection-algorithm/SSD
详细文档说明:SSD