mmyolo/configs/ppyoloe/metafile.yml
Nioolek 8127805dd3
[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

70 lines
2.4 KiB
YAML

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