detection_template
一个目标检测的通用框架(不需要cuda编译),支持Yolo全系列(v2~v5)、EfficientDet、RetinaNet、Cascade-RCNN等SOTA网络,Kaggle小麦检测12/2245。
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 | 数据集 | 论文[email protected] | 复现[email protected] | 下载链接 | 密码 | 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
-
YoloV2、YoloV3 https://github.com/andy-yun/pytorch-0.4-yolov3
-
EfficientDet https://github.com/rwightman/efficientdet-pytorch
-
YoloV4 https://github.com/Tianxiaomo/pytorch-YOLOv4 https://github.com/argusswift/YOLOv4-pytorch
-
Faster_RCNN https://github.com/pytorch/vision/tree/master/torchvision/models/detection