mirror of https://github.com/open-mmlab/mmyolo.git
315 lines
12 KiB
YAML
315 lines
12 KiB
YAML
Collections:
|
|
- Name: YOLOv5
|
|
Metadata:
|
|
Training Data: COCO
|
|
Training Techniques:
|
|
- SGD with Nesterov
|
|
- Weight Decay
|
|
- AMP
|
|
- Synchronize BN
|
|
Training Resources: 8x A100 GPUs
|
|
Architecture:
|
|
- CSPDarkNet
|
|
- PAFPN
|
|
README: configs/yolov5/README.md
|
|
Code:
|
|
URL: https://github.com/open-mmlab/mmyolo/blob/v0.1.0/mmyolo/models/detectors/yolo_detector.py#L12
|
|
Version: v0.1.0
|
|
- Name: YOLOv5_VOC
|
|
Metadata:
|
|
Training Data: VOC
|
|
Training Techniques:
|
|
- SGD with Nesterov
|
|
- Weight Decay
|
|
- AMP
|
|
Training Resources: 1x A100 GPU
|
|
Architecture:
|
|
- CSPDarkNet
|
|
- PAFPN
|
|
README: configs/yolov5/README.md
|
|
Code:
|
|
URL: https://github.com/open-mmlab/mmyolo/blob/v0.1.0/mmyolo/models/detectors/yolo_detector.py#L12
|
|
Version: v0.1.0
|
|
|
|
Models:
|
|
- Name: yolov5_n-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_n-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 1.5
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 28.0
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_n-v61_syncbn_fast_8xb16-300e_coco/yolov5_n-v61_syncbn_fast_8xb16-300e_coco_20220919_090739-b804c1ad.pth
|
|
- Name: yolov5_s-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_s-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 2.7
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 37.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_s-v61_syncbn_fast_8xb16-300e_coco/yolov5_s-v61_syncbn_fast_8xb16-300e_coco_20220918_084700-86e02187.pth
|
|
- Name: yolov5_m-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_m-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 5.0
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 45.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_m-v61_syncbn_fast_8xb16-300e_coco/yolov5_m-v61_syncbn_fast_8xb16-300e_coco_20220917_204944-516a710f.pth
|
|
- Name: yolov5_l-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_l-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 8.1
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 48.8
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_l-v61_syncbn_fast_8xb16-300e_coco/yolov5_l-v61_syncbn_fast_8xb16-300e_coco_20220917_031007-096ef0eb.pth
|
|
- Name: yolov5_x-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_x-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 12.2
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 50.2
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_x-v61_syncbn_fast_8xb16-300e_coco/yolov5_x-v61_syncbn_fast_8xb16-300e_coco_20230305_152943-00776a4b.pth
|
|
- Name: yolov5_n-p6-v62_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_n-p6-v62_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 5.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 35.9
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_n-p6-v62_syncbn_fast_8xb16-300e_coco/yolov5_n-p6-v62_syncbn_fast_8xb16-300e_coco_20221027_224705-d493c5f3.pth
|
|
- Name: yolov5_s-p6-v62_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_s-p6-v62_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 10.5
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 44.4
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_s-p6-v62_syncbn_fast_8xb16-300e_coco/yolov5_s-p6-v62_syncbn_fast_8xb16-300e_coco_20221027_215044-58865c19.pth
|
|
- Name: yolov5_m-p6-v62_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_m-p6-v62_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 19.1
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 51.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_m-p6-v62_syncbn_fast_8xb16-300e_coco/yolov5_m-p6-v62_syncbn_fast_8xb16-300e_coco_20221027_230453-49564d58.pth
|
|
- Name: yolov5_l-p6-v62_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/yolov5_l-p6-v62_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 30.5
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 53.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_l-p6-v62_syncbn_fast_8xb16-300e_coco/yolov5_l-p6-v62_syncbn_fast_8xb16-300e_coco_20221027_234308-7a2ba6bf.pth
|
|
- Name: yolov5_n-v61_fast_1xb64-50e_voc
|
|
In Collection: YOLOv5_VOC
|
|
Config: configs/yolov5/voc/yolov5_n-v61_fast_1xb64-50e_voc.py
|
|
Metadata:
|
|
Training Memory (GB): 3.5
|
|
Epochs: 50
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 51.2
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_n-v61_fast_1xb64-50e_voc/yolov5_n-v61_fast_1xb64-50e_voc_20221017_234254-f1493430.pth
|
|
- Name: yolov5_s-v61_fast_1xb64-50e_voc
|
|
In Collection: YOLOv5_VOC
|
|
Config: configs/yolov5/voc/yolov5_s-v61_fast_1xb64-50e_voc.py
|
|
Metadata:
|
|
Training Memory (GB): 6.5
|
|
Epochs: 50
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 62.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_s-v61_fast_1xb64-50e_voc/yolov5_s-v61_fast_1xb64-50e_voc_20221017_234156-0009b33e.pth
|
|
- Name: yolov5_m-v61_fast_1xb64-50e_voc
|
|
In Collection: YOLOv5_VOC
|
|
Config: configs/yolov5/voc/yolov5_m-v61_fast_1xb64-50e_voc.py
|
|
Metadata:
|
|
Training Memory (GB): 12.0
|
|
Epochs: 50
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 70.1
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_m-v61_fast_1xb64-50e_voc/yolov5_m-v61_fast_1xb64-50e_voc_20221017_114138-815c143a.pth
|
|
- Name: yolov5_l-v61_fast_1xb32-50e_voc
|
|
In Collection: YOLOv5_VOC
|
|
Config: configs/yolov5/voc/yolov5_l-v61_fast_1xb32-50e_voc.py
|
|
Metadata:
|
|
Training Memory (GB): 10.0
|
|
Epochs: 50
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 73.1
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_l-v61_fast_1xb32-50e_voc/yolov5_l-v61_fast_1xb32-50e_voc_20221017_045500-edc7e0d8.pth
|
|
- Name: yolov5_n_mask-refine-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/mask_refine/yolov5_n_mask-refine-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 1.5
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 28.0
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/mask_refine/yolov5_n_mask-refine-v61_syncbn_fast_8xb16-300e_coco/yolov5_n_mask-refine-v61_syncbn_fast_8xb16-300e_coco_20230305_152706-712fb1b2.pth
|
|
- Name: yolov5_s_mask-refine-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/mask_refine/yolov5_s_mask-refine-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 2.7
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 38.0
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/mask_refine/yolov5_s_mask-refine-v61_syncbn_fast_8xb16-300e_coco/yolov5_s_mask-refine-v61_syncbn_fast_8xb16-300e_coco_20230304_033134-8e0cd271.pth
|
|
- Name: yolov5_m_mask-refine-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/mask_refine/yolov5_m_mask-refine-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 5.0
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 45.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/mask_refine/yolov5_m_mask-refine-v61_syncbn_fast_8xb16-300e_coco/yolov5_m_mask-refine-v61_syncbn_fast_8xb16-300e_coco_20230305_153946-44e96155.pth
|
|
- Name: yolov5_l_mask-refine-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/mask_refine/yolov5_l_mask-refine-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 8.1
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 49.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/mask_refine/yolov5_l_mask-refine-v61_syncbn_fast_8xb16-300e_coco/yolov5_l_mask-refine-v61_syncbn_fast_8xb16-300e_coco_20230305_154301-2c1d912a.pth
|
|
- Name: yolov5_x_mask-refine-v61_syncbn_fast_8xb16-300e_coco
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/mask_refine/yolov5_x_mask-refine-v61_syncbn_fast_8xb16-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 12.2
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 50.9
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/mask_refine/yolov5_x_mask-refine-v61_syncbn_fast_8xb16-300e_coco/yolov5_x_mask-refine-v61_syncbn_fast_8xb16-300e_coco_20230305_154321-07edeb62.pth
|
|
- Name: yolov5_ins_n-v61_syncbn_fast_8xb16-300e_coco_instance
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/ins_seg/yolov5_ins_n-v61_syncbn_fast_8xb16-300e_coco_instance.py
|
|
Metadata:
|
|
Training Memory (GB): 3.3
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 27.9
|
|
- Task: Instance Segmentation
|
|
Dataset: COCO
|
|
Metrics:
|
|
mask AP: 23.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/ins_seg/yolov5_ins_n-v61_syncbn_fast_8xb16-300e_coco_instance/yolov5_ins_n-v61_syncbn_fast_8xb16-300e_coco_instance_20230424_104807-84cc9240.pth
|
|
- Name: yolov5_ins_s-v61_syncbn_fast_8xb16-300e_coco_instance
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/ins_seg/yolov5_ins_s-v61_syncbn_fast_8xb16-300e_coco_instance.py
|
|
Metadata:
|
|
Training Memory (GB): 4.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 38.1
|
|
- Task: Instance Segmentation
|
|
Dataset: COCO
|
|
Metrics:
|
|
mask AP: 32.0
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/ins_seg/yolov5_ins_s-v61_syncbn_fast_8xb16-300e_coco_instance/yolov5_ins_s-v61_syncbn_fast_8xb16-300e_coco_instance_20230426_012542-3e570436.pth
|
|
- Name: yolov5_ins_s-v61_syncbn_fast_non_overlap_8xb16-300e_coco_instance
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/ins_seg/yolov5_ins_s-v61_syncbn_fast_non_overlap_8xb16-300e_coco_instance.py
|
|
Metadata:
|
|
Training Memory (GB): 4.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 38.0
|
|
- Task: Instance Segmentation
|
|
Dataset: COCO
|
|
Metrics:
|
|
mask AP: 32.1
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/ins_seg/yolov5_ins_s-v61_syncbn_fast_non_overlap_8xb16-300e_coco_instance/yolov5_ins_s-v61_syncbn_fast_non_overlap_8xb16-300e_coco_instance_20230424_104642-6780d34e.pth
|
|
- Name: yolov5_ins_m-v61_syncbn_fast=_8xb16-300e_coco_instance
|
|
In Collection: YOLOv5
|
|
Config: configs/yolov5/ins_seg/yolov5_ins_m-v61_syncbn_fast=_8xb16-300e_coco_instance.py
|
|
Metadata:
|
|
Training Memory (GB): 7.3
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 45.1
|
|
- Task: Instance Segmentation
|
|
Dataset: COCO
|
|
Metrics:
|
|
mask AP: 37.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolov5/ins_seg/yolov5_ins_m-v61_syncbn_fast_8xb16-300e_coco_instance/yolov5_ins_m-v61_syncbn_fast_8xb16-300e_coco_instance_20230424_111529-ef5ba1a9.pth
|