YOLO_v1

实现YOLO_v1目标检测算法

实现流程

  1. 创建训练数据集
  2. 定义YOLO_v1模型
  3. 定义损失函数Multi-part Loss
  4. 训练模型
  5. 计算mAP

训练结果

  • mAP
    • 对于Location DataSet, 50轮训练完成后能够实现97.31% mAP
    • 对于VOC 07(train+val)50轮训练完成后能够实现75.25% mAP

相关文档