目标检测训练框架

研究lufficc/SSD源码时,发现其实现了一个目标检测训练框架,进行深入研究后整理成一个新的训练框架

框架概述

将目标检测训练框架分为以下内容:

  1. 配置模块
  2. 日志模块
  3. 数据模块
  4. 检测器模块
  5. 优化器模块
  6. 辅助模块

该训练框架能够提供以下功能:

  • 功能模块化:适用于不同算法的替换和移植
  • 配置规范化:统一配置训练参数
  • 日志读写:详细打印训练和测试结果,包含了文本日志和可视化日志
  • 可持续训练:通过定期检查点保存,实现中断训练后的恢复训练

框架实现

模块实现地址:object-detection-algorithm/SSD

详细文档说明:SSD