mmyolo/configs/ppyoloe/metafile.yml

70 lines
2.4 KiB
YAML
Raw Normal View History

[Feature] Support PPYOLOE training (#259) * add ppyoloe backbone, neck * add ppyoloe test * add docstring * add ppyoloe m/l/x configfile * add ppyoloe_coco.py * rename config * add typehint * format code; add ut * add datapre * add datapre * add ppyoloe datapre * add ppyoloe datapre * add ppyoloe datapre * reproduce coco v0.1 * add ut * add ut, docstring * fix transforms bug * use mmdet dfloss * add non plus model config * add non plus model config * fix * add ut * produce coco v0.2 * fix config * fix config * fix eps and transforms bug * add ema * fix resize * fix transforms.py * fix transforms.py * fix transforms.py * old version * old version * old version * old version * old version * old version * fix stride loss error * add INTER_LANCZOS4 * fix crop bug * init commit * format code * format code * bgr transforms.py * add typehint and doc in transforms.py * 继承新版yolov6head写法,删除不必要的注释 * fix transforms var name bug * bbox decode use stridetensor insted of priors * add headmodule todo * add ppyoloe README.md * add ppyoloe README.md * Update tests/test_datasets/test_transforms/test_transforms.py Co-authored-by: Range King <RangeKingHZ@gmail.com> * Update tests/test_datasets/test_transforms/test_transforms.py Co-authored-by: Range King <RangeKingHZ@gmail.com> * save ckpt last 10 epochs * save_best ckpt * del ppyoloe collate * change name of ppyoloebatchrandomresize * add the reason for rewritten PPYOLOEDetDataPreprocessor * solve ppyoloerandomresize name error * rm PPYOLOERandomExpand * rm l1 loss * rm yolov6 loss_obj * add difference between yolov6 and ppyoloe * add reason for rewrite paramscheduler * change proj init way * fix error * rm proj_conv in pth * format code * add load_from * update * support fast training * add pretrained model url * update * add pretrained model url * fix error * add imagenet model convert and use init_cfg to init backbone * add plus model pretrain model * add ut * add ut * fix ut * fix withstride bug * cat in yolov5_collate * merge * fix typehint * update readme * add reason for gap * fix log in README.md * rollback yolov6 * change inherit * fix ut * fix ut Co-authored-by: Range King <RangeKingHZ@gmail.com> Co-authored-by: hha <1286304229@qq.com> Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-06 15:54:39 +08:00
Collections:
- Name: PPYOLOE
Metadata:
Training Data: COCO
Training Techniques:
- SGD with Nesterov
- Weight Decay
- Synchronize BN
Training Resources: 8x A100 GPUs
Architecture:
- PPYOLOECSPResNet
- PPYOLOECSPPAFPN
Paper:
URL: https://arxiv.org/abs/2203.16250
Title: 'PP-YOLOE: An evolved version of YOLO'
README: configs/ppyoloe/README.md
Code:
URL: https://github.com/open-mmlab/mmyolo/blob/v0.0.1/mmyolo/models/detectors/yolo_detector.py#L12
Version: v0.0.1
Models:
- Name: ppyoloe_plus_s_fast_8xb8-80e_coco
In Collection: PPYOLOE
Config: configs/ppyoloe/ppyoloe_plus_s_fast_8xb8-80e_coco.py
Metadata:
Training Memory (GB): 4.7
Epochs: 80
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 43.5
Weights: https://download.openmmlab.com/mmyolo/v0/ppyoloe/ppyoloe_plus_s_fast_8xb8-80e_coco/ppyoloe_plus_s_fast_8xb8-80e_coco_20230101_154052-9fee7619.pth
- Name: ppyoloe_plus_m_fast_8xb8-80e_coco
In Collection: PPYOLOE
Config: configs/ppyoloe/ppyoloe_plus_m_fast_8xb8-80e_coco.py
Metadata:
Training Memory (GB): 8.4
Epochs: 80
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 49.5
Weights: https://download.openmmlab.com/mmyolo/v0/ppyoloe/ppyoloe_plus_m_fast_8xb8-80e_coco/ppyoloe_plus_m_fast_8xb8-80e_coco_20230104_193132-e4325ada.pth
- Name: ppyoloe_plus_L_fast_8xb8-80e_coco
In Collection: PPYOLOE
Config: configs/ppyoloe/ppyoloe_plus_L_fast_8xb8-80e_coco.py
Metadata:
Training Memory (GB): 13.2
Epochs: 80
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 52.6
Weights: https://download.openmmlab.com/mmyolo/v0/ppyoloe/ppyoloe_plus_l_fast_8xb8-80e_coco/ppyoloe_plus_l_fast_8xb8-80e_coco_20230102_203825-1864e7b3.pth
- Name: ppyoloe_plus_x_fast_8xb8-80e_coco
In Collection: PPYOLOE
Config: configs/ppyoloe/ppyoloe_plus_x_fast_8xb8-80e_coco.py
Metadata:
Training Memory (GB): 19.1
Epochs: 80
Results:
- Task: Object Detection
Dataset: COCO
Metrics:
box AP: 54.2
Weights: https://download.openmmlab.com/mmyolo/v0/ppyoloe/ppyoloe_plus_x_fast_8xb8-80e_coco/ppyoloe_plus_x_fast_8xb8-80e_coco_20230104_194921-8c953949.pth