/detection_template

一个目标检测的通用框架(不需要cuda编译)，支持Yolo全系列(v2~v5)、EfficientDet、RetinaNet、Cascade-RCNN等SOTA网络。

Primary LanguagePythonMIT LicenseMIT

detection_template

一个目标检测的通用框架(不需要cuda编译)，支持Yolo全系列(v2~v5)、EfficientDet、RetinaNet、Cascade-RCNN等SOTA网络，Kaggle小麦检测12/2245。

preview

Functionalities

  • 数据格式

    • VOC
    • COCO
    • 自定义格式

  • 网络模型

    • EfficientDet(目前不支持训练过程中验证)
    • YoloV2、V3
    • YoloV4
    • YoloV5
    • SSD300、SSD512
    • Faster-RCNN
    • Cascade-RCNN
    • RetinaNet

  • TTA

    • 多尺度融合
    • nms
    • Weighted Box Fusion(WBF)
    • 伪标签

  • Scheduler

    • Step Scheduler
    • 验证集指标不下降时学习率衰减

  • Metrics

    • mAP

  • 可视化

    • 数据集bbox预览
    • dataloader数据增强预览
    • 预测结果预览

  • 辅助工具

    • 手工标注工具

安装和使用教程

安装和使用教程见 get_started.md.

预训练模型

Model backbone 数据集 论文mAP@.5 复现mAP@.5 下载链接 密码 sha256
YoloV2 Darknet-19 VOC 76.8 76.46 [百度网盘] mwik 5d29a34b
YoloV3 Darknet-19 COCO 55.3 - [百度网盘] cf4j 943b926a
FRCNN Res50+FPN VOC - 83.26 [百度网盘] isqt 3d5c3b15
FRCNN Res50+FPN COCO 48.81 [pytorch] - 258fb6c6
YoloV4 CSPDarknet-53 COCO 62.8 - [百度网盘] nio7 797dc954
YoloV5 CSPDarknet COCO - 64.30 [百度网盘] cssw 8e54a2e8
SSD300 VGG16 VOC 77.2 79.21 [百度网盘] 59y0 106c0fc9
SSD512 VGG16 VOC 79.8 82.14 [百度网盘] 0iur 844b40b3

Reference