mirror of
https://github.com/open-mmlab/mmyolo.git
synced 2025-06-03 15:00:20 +08:00
6ecebdbbd5
* add * reproduce map * add typehint and doc * format code * replace key * add ut * format * format * format code * fix ut * fix ut * fix comment * fix comment * fix comment * [WIP][Feature] Support yolov5-Ins training * fix comment * change data flow and fix loss_mask compute * align the data pipeline * remove albu gt mask key * support yolov5 ins inference * fix multi gpu test * align the post_process with v8 * support training * support training * code formatting * code formatting * Support pad_param type (#672) * add half_pad_param * fix default fast_test * fix loss weight compute * add models * add dataset1 * add dataset2 * add dataset3 * add configs * re commit __init__ * re commit __init__ * re commit * del local * add typo * del PoseToDetConverter and BBoxKeypoints * del local changes * fix mask rescale, add segment merge, fix segment2bbox * fix pipeline * add dataset * fix typo * add resize in mmyolo * fix typo * del local * del local changes * del local changes * fix dir name * fix dir name * add FilterAnnotations * fix typo * new config for yolox-pose * fix typo * fix typo * fix clip and fix mask init * del pose dataset changes * fix YOLOv5DetDataPreprocessor * del local file * fix typo * del init_cfg * simplify config * fix batch size * fix batch size * fix typo * code formatting * code formatting * code formatting * code formatting * fix bug for FilterAnnotations * simpler way for FilterAnnotations * update config * [Fix] fix load image from file * shorten eval time * fix typo * add large model * [Add] Add docs and more config * [Fix] config type and test_formatting * [Fix] fix yolov5-ins_m packdetinputs * hand rebase from yolov5-ins * use new PackDetInputs * rebase fix typo * add mapping table * fix typo * add weight * del typo * del typo * add results * install mmpose, Keypoints note, context manager, predict, ota rename * fix test * add unittest for pose_sim_ota_assigner and yolox_head * add unittest for pose_sim_ota_assigner and yolox_head * fix typo --------- Co-authored-by: Nioolek <379319054@qq.com> Co-authored-by: josonchan <josonchan1998@163.com> Co-authored-by: Nioolek <40284075+Nioolek@users.noreply.github.com> Co-authored-by: huanghaian <huanghaian@sensetime.com>
167 lines
6.0 KiB
YAML
167 lines
6.0 KiB
YAML
Collections:
|
|
- Name: YOLOX
|
|
Metadata:
|
|
Training Data: COCO
|
|
Training Techniques:
|
|
- SGD with Nesterov
|
|
- Weight Decay
|
|
- Cosine Annealing Lr Updater
|
|
Training Resources: 8x A100 GPUs
|
|
Architecture:
|
|
- CSPDarkNet
|
|
- PAFPN
|
|
Paper:
|
|
URL: https://arxiv.org/abs/2107.08430
|
|
Title: 'YOLOX: Exceeding YOLO Series in 2021'
|
|
README: configs/yolox/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: yolox_tiny_fast_8xb8-300e_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_tiny_fast_8xb8-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 2.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 32.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_tiny_8xb8-300e_coco/yolox_tiny_8xb8-300e_coco_20220919_090908-0e40a6fc.pth
|
|
- Name: yolox_s_fast_8xb8-300e_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_s_fast_8xb8-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 2.9
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 40.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_s_fast_8xb8-300e_coco/yolox_s_fast_8xb8-300e_coco_20230213_142600-2b224d8b.pth
|
|
- Name: yolox_m_fast_8xb8-300e_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_m_fast_8xb8-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 4.9
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 46.9
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_m_fast_8xb8-300e_coco/yolox_m_fast_8xb8-300e_coco_20230213_160218-a71a6b25.pth
|
|
- Name: yolox_l_fast_8xb8-300e_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_l_fast_8xb8-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 8.0
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 50.1
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_l_fast_8xb8-300e_coco/yolox_l_fast_8xb8-300e_coco_20230213_160715-c731eb1c.pth
|
|
- Name: yolox_x_fast_8xb8-300e_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_x_fast_8xb8-300e_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 9.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 51.4
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_x_fast_8xb8-300e_coco/yolox_x_fast_8xb8-300e_coco_20230215_133950-1d509fab.pth
|
|
- Name: yolox_tiny_fast_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_tiny_fast_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 4.9
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 34.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_tiny_fast_8xb32-300e-rtmdet-hyp_coco/yolox_tiny_fast_8xb32-300e-rtmdet-hyp_coco_20230210_143637-4c338102.pth
|
|
- Name: yolox_s_fast_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_s_fast_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 9.8
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 41.9
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_s_fast_8xb32-300e-rtmdet-hyp_coco/yolox_s_fast_8xb32-300e-rtmdet-hyp_coco_20230210_134645-3a8dfbd7.pth
|
|
- Name: yolox_m_fast_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: configs/yolox/yolox_m_fast_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 17.6
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Object Detection
|
|
Dataset: COCO
|
|
Metrics:
|
|
box AP: 47.5
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/yolox_m_fast_8xb32-300e-rtmdet-hyp_coco/yolox_m_fast_8xb32-300e-rtmdet-hyp_coco_20230210_144328-e657e182.pth
|
|
- Name: yolox-pose_tiny_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: yolox-pose_tiny_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 5.3
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Human Pose Estimation
|
|
Dataset: COCO
|
|
Metrics:
|
|
AP: 52.8
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/pose/yolox-pose_tiny_8xb32-300e-rtmdet-hyp_coco/yolox-pose_tiny_8xb32-300e-rtmdet-hyp_coco_20230427_080351-2117af67.pth
|
|
- Name: yolox-pose_s_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: yolox-pose_s_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 10.7
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Human Pose Estimation
|
|
Dataset: COCO
|
|
Metrics:
|
|
AP: 63.7
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/pose/yolox-pose_s_8xb32-300e-rtmdet-hyp_coco/yolox-pose_s_8xb32-300e-rtmdet-hyp_coco_20230427_005150-e87d843a.pth
|
|
- Name: yolox-pose_m_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: yolox-pose_m_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 19.2
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Human Pose Estimation
|
|
Dataset: COCO
|
|
Metrics:
|
|
AP: 69.3
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/pose/yolox-pose_m_8xb32-300e-rtmdet-hyp_coco/yolox-pose_m_8xb32-300e-rtmdet-hyp_coco_20230427_094024-bbeacc1c.pth
|
|
- Name: yolox-pose_l_8xb32-300e-rtmdet-hyp_coco
|
|
In Collection: YOLOX
|
|
Config: yolox-pose_l_8xb32-300e-rtmdet-hyp_coco.py
|
|
Metadata:
|
|
Training Memory (GB): 30.3
|
|
Epochs: 300
|
|
Results:
|
|
- Task: Human Pose Estimation
|
|
Dataset: COCO
|
|
Metrics:
|
|
AP: 71.1
|
|
Weights: https://download.openmmlab.com/mmyolo/v0/yolox/pose/yolox-pose_l_8xb32-300e-rtmdet-hyp_coco/yolox-pose_l_8xb32-300e-rtmdet-hyp_coco_20230427_041140-82d65ac8.pth
|