mirror of https://github.com/open-mmlab/mmyolo.git
69 lines
2.2 KiB
YAML
69 lines
2.2 KiB
YAML
Collections:
|
|
- Name: RTMDet
|
|
Metadata:
|
|
Training Data: COCO
|
|
Training Techniques:
|
|
- AdamW
|
|
- Flat Cosine Annealing
|
|
Training Resources: 8x A100 GPUs
|
|
Architecture:
|
|
- CSPNeXt
|
|
- CSPNeXtPAFPN
|
|
README: configs/rtmdet/README.md
|
|
Code:
|
|
URL: https://github.com/open-mmlab/mmyolo/blob/main/mmyolo/models/detectors/yolo_detector.py#L12
|
|
Version: v0.1.1
|
|
|
|
Models:
|
|
- Name: rtmdet_tiny_syncbn_8xb32-300e_coco
|
|
In Collection: RTMDet
|
|
Config: configs/rtmdet/rtmdet_tiny_syncbn_8xb32-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 11.7
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 40.9
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/rtmdet/rtmdet_tiny_syncbn_8xb32-300e_coco/rtmdet_tiny_syncbn_8xb32-300e_coco_20220902_112414-259f3241.pth
|
|
|
|
- Name: rtmdet_s_syncbn_8xb32-300e_coco
|
|
In Collection: RTMDet
|
|
Config: configs/rtmdet/rtmdet_s_syncbn_8xb32-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 15.9
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 44.5
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/rtmdet/rtmdet_s_syncbn_8xb32-300e_coco/rtmdet_s_syncbn_8xb32-300e_coco_20220905_161602-fd1cacb9.pth
|
|
|
|
- Name: rtmdet_m_syncbn_8xb32-300e_coco
|
|
In Collection: RTMDet
|
|
Config: configs/rtmdet/rtmdet_m_syncbn_8xb32-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 27.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 49.1
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/rtmdet/rtmdet_m_syncbn_8xb32-300e_coco/rtmdet_m_syncbn_8xb32-300e_coco_20220924_132959-d9f2e90d.pth
|
|
|
|
- Name: rtmdet_l_syncbn_8xb32-300e_coco
|
|
In Collection: RTMDet
|
|
Config: configs/rtmdet/rtmdet_l_syncbn_8xb32-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 43.2
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 51.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/rtmdet/rtmdet_l_syncbn_8xb32-300e_coco/rtmdet_l_syncbn_8xb32-300e_coco_20220926_150401-40c754b5.pth
|