Skip to content

Ppyoloe train paddle #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 27 commits into
base: ppyoloe_train
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
7c5e21c
Fix customize dataset training (#428)
triple-Mu Jan 4, 2023
28439f1
[Deploy] Support YOLOv7 easy deploy (#427)
triple-Mu Jan 4, 2023
e146fbe
add official version ppyoloe
Nioolek Jan 5, 2023
4d5db0d
add official version ppyoloe
Nioolek Jan 5, 2023
88c75ba
add plus x config
Nioolek Jan 5, 2023
d4ebf6d
fix
Nioolek Jan 5, 2023
cac29de
save_interval =5
Nioolek Jan 5, 2023
5172805
fix rtmdet-s cfg and neck (#434)
RangiLyu Jan 6, 2023
d8cbf21
[Fix] Fix typo in docstring (#433)
MambaWong Jan 6, 2023
2290713
[Fix] Fix bugs in the output annotation file when `--class-id-txt` is…
VoyagerXvoyagerx Jan 6, 2023
fa80011
add s 300e config
Nioolek Jan 6, 2023
2f399df
[Feature] Add the unit test to test the `scale_factor` in LetterResiz…
tang576225574 Jan 6, 2023
c14ecc3
[DOC] Update useful_tools.md (#384)
matrixgame2018 Jan 6, 2023
a20b160
[CodeCamp]Add doc of yolov6_description.md (#382)
wzr-skn Jan 6, 2023
8127805
[Feature] Support PPYOLOE training (#259)
Nioolek Jan 6, 2023
8843301
[BC Breaking] Keep the same input shape order as mmdet 3.0.0rc5 (#436)
hhaAndroid Jan 6, 2023
07afc3e
Fix potential bugs in rtmdet multi-scale training (#439)
Nioolek Jan 6, 2023
7d07e1a
Update yolo_detector.py (#441)
vaew Jan 6, 2023
89275ad
[CodeCamp] #50 Add module combination doc (#349)
Seperendity Jan 6, 2023
a631370
Update model graphs (#442)
RangeKing Jan 7, 2023
fc107b8
[Docs] Add an example project to 'projects\' folder (#401)
dsghaonan Jan 8, 2023
9c6b4fe
Add v0.3.0 changelog (#440)
hhaAndroid Jan 8, 2023
b0cf073
Bump version to v0.3.0
hhaAndroid Jan 8, 2023
c682ae8
adjust the minimum supported python version to 3.9 (#449)
tianleiSHI Jan 9, 2023
eb2634d
merge
Nioolek Jan 10, 2023
a72e713
add m config
Nioolek Jan 10, 2023
47eb4c1
add m config
Nioolek Jan 10, 2023
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 4 additions & 6 deletions .circleci/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ jobs:
type: string
cuda:
type: enum
enum: ["10.1", "10.2", "11.1","11.0"]
enum: ["10.1", "10.2", "11.1", "11.0"]
cudnn:
type: integer
default: 7
Expand Down Expand Up @@ -151,8 +151,7 @@ workflows:

pr_stage_test:
when:
not:
<< pipeline.parameters.lint_only >>
not: << pipeline.parameters.lint_only >>
jobs:
- lint:
name: lint
Expand All @@ -164,7 +163,7 @@ workflows:
name: minimum_version_cpu
torch: 1.8.0
torchvision: 0.9.0
python: 3.8.0 # The lowest python 3.6.x version available on CircleCI images
python: 3.8.0 # The lowest python 3.7.x version available on CircleCI images
requires:
- lint
- build_cpu:
Expand All @@ -188,8 +187,7 @@ workflows:
- hold
merge_stage_test:
when:
not:
<< pipeline.parameters.lint_only >>
not: << pipeline.parameters.lint_only >>
jobs:
- build_cuda:
name: minimum_version_gpu
Expand Down
40 changes: 31 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -71,14 +71,32 @@ And the figure of P6 model is in [model_design.md](docs/en/algorithm_description

## What's New

💎 **v0.2.0** was released on 1/12/2022:
### Highlight

1. Support [YOLOv7](https://github.com/open-mmlab/mmyolo/tree/dev/configs/yolov7) P5 and P6 model
2. Support [YOLOv6](https://github.com/open-mmlab/mmyolo/blob/dev/configs/yolov6/README.md) ML model
3. Support [Grad-Based CAM and Grad-Free CAM](https://github.com/open-mmlab/mmyolo/blob/dev/demo/boxam_vis_demo.py)
4. Support [large image inference](https://github.com/open-mmlab/mmyolo/blob/dev/demo/large_image_demo.py) based on sahi
5. Add [easydeploy](https://github.com/open-mmlab/mmyolo/blob/dev/projects/easydeploy/README.md) project under the projects folder
6. Add [custom dataset guide](https://github.com/open-mmlab/mmyolo/blob/dev/docs/zh_cn/user_guides/custom_dataset.md)
We are excited to announce our latest work on real-time object recognition tasks, **RTMDet**, a family of fully convolutional single-stage detectors. RTMDet not only achieves the best parameter-accuracy trade-off on object detection from tiny to extra-large model sizes but also obtains new state-of-the-art performance on instance segmentation and rotated object detection tasks. Details can be found in the [technical report](https://arxiv.org/abs/2212.07784). Pre-trained models are [here](configs/rtmdet).

[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/rtmdet-an-empirical-study-of-designing-real/real-time-instance-segmentation-on-mscoco)](https://paperswithcode.com/sota/real-time-instance-segmentation-on-mscoco?p=rtmdet-an-empirical-study-of-designing-real)
[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/rtmdet-an-empirical-study-of-designing-real/object-detection-in-aerial-images-on-dota-1)](https://paperswithcode.com/sota/object-detection-in-aerial-images-on-dota-1?p=rtmdet-an-empirical-study-of-designing-real)
[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/rtmdet-an-empirical-study-of-designing-real/object-detection-in-aerial-images-on-hrsc2016)](https://paperswithcode.com/sota/object-detection-in-aerial-images-on-hrsc2016?p=rtmdet-an-empirical-study-of-designing-real)

| Task | Dataset | AP | FPS(TRT FP16 BS1 3090) |
| ------------------------ | ------- | ------------------------------------ | ---------------------- |
| Object Detection | COCO | 52.8 | 322 |
| Instance Segmentation | COCO | 44.6 | 188 |
| Rotated Object Detection | DOTA | 78.9(single-scale)/81.3(multi-scale) | 121 |

<div align=center>
<img src="https://user-images.githubusercontent.com/12907710/208044554-1e8de6b5-48d8-44e4-a7b5-75076c7ebb71.png"/>
</div>

MMYOLO currently only implements the object detection algorithm, but it has a significant training acceleration compared to the MMDeteciton version. The training speed is 2.6 times faster than the previous version.

💎 **v0.3.0** was released on 8/1/2023:

1. Implement fast version of [RTMDet](https://github.com/open-mmlab/mmyolo/blob/dev/configs/rtmdet/README.md). RTMDet-s 8xA100 training takes only 14 hours. The training speed is 2.6 times faster than the previous version.
2. Support [PPYOLOE](https://github.com/open-mmlab/mmyolo/blob/dev/configs/ppyoloe/README.md) training
3. Support `iscrowd` attribute training in [YOLOv5](https://github.com/open-mmlab/mmyolo/blob/dev/configs/yolov5/crowdhuman/yolov5_s-v61_8xb16-300e_ignore_crowdhuman.py)
4. Support [YOLOv5 assigner result visualization](https://github.com/open-mmlab/mmyolo/blob/dev/projects/assigner_visualization/README.md)

For release history and update details, please refer to [changelog](https://mmyolo.readthedocs.io/en/latest/notes/changelog.html).

Expand All @@ -92,7 +110,7 @@ conda activate open-mmlab
pip install openmim
mim install "mmengine>=0.3.1"
mim install "mmcv>=2.0.0rc1,<2.1.0"
mim install "mmdet>=3.0.0rc3,<3.1.0"
mim install "mmdet>=3.0.0rc5,<3.1.0"
git clone https://github.com/open-mmlab/mmyolo.git
cd mmyolo
# Install albumentations
Expand Down Expand Up @@ -152,7 +170,7 @@ Results and models are available in the [model zoo](docs/en/model_zoo.md).
- [x] [RTMDet](configs/rtmdet)
- [x] [YOLOv6](configs/yolov6)
- [x] [YOLOv7](configs/yolov7)
- [ ] [PPYOLOE](configs/ppyoloe)(Inference only)
- [x] [PPYOLOE](configs/ppyoloe)

</details>

Expand Down Expand Up @@ -183,6 +201,8 @@ Results and models are available in the [model zoo](docs/en/model_zoo.md).
<li>YOLOXCSPDarknet</li>
<li>EfficientRep</li>
<li>CSPNeXt</li>
<li>YOLOv7Backbone</li>
<li>PPYOLOECSPResNet</li>
</ul>
</td>
<td>
Expand All @@ -191,6 +211,8 @@ Results and models are available in the [model zoo](docs/en/model_zoo.md).
<li>YOLOv6RepPAFPN</li>
<li>YOLOXPAFPN</li>
<li>CSPNeXtPAFPN</li>
<li>YOLOv7PAFPN</li>
<li>PPYOLOECSPPAFPN</li>
</ul>
</td>
<td>
Expand Down
Loading