Skip to content

Commit

Permalink
update readme for ms2.3.1
Browse files Browse the repository at this point in the history
  • Loading branch information
WongGawa committed Nov 6, 2024
1 parent 26f1b76 commit 687d0ae
Show file tree
Hide file tree
Showing 10 changed files with 53 additions and 11 deletions.
2 changes: 1 addition & 1 deletion GETTING_STARTED.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ to understand their behavior. Some common arguments are:
* Prepare your dataset in YOLO format. If trained with COCO (YOLO format), prepare it from [yolov5](https://github.com/ultralytics/yolov5) or the darknet.

<details onclose>

<summary><b>View More</b></summary>
```
coco/
{train,val}2017.txt
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ The following is the corresponding `mindyolo` versions and supported `mindspore`
| mindyolo | mindspore |
| :--: | :--: |
| master | master |
| 0.4 | 2.3.0 |
| 0.4 | 2.3.1/2.3.0 |
| 0.3 | 2.2.10 |
| 0.2 | 2.0 |
| 0.1 | 1.8 |
Expand Down
21 changes: 14 additions & 7 deletions configs/yolov3/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,17 +14,17 @@ We present some updates to YOLO! We made a bunch of little design changes to mak
<details open markdown>
<summary><b>performance tested on Ascend 910(8p) with graph mode</b></summary>

| Name | Scale | BatchSize | ImageSize | Dataset | Box mAP (%) | Params | Recipe | Download |
|--------| :---: | :---: | :---: |--------------| :---: | :---: | :---: | :---: |
| YOLOv3 | Darknet53 | 16 * 8 | 640 | MS COCO 2017 | 45.5 | 61.9M | [yaml](./yolov3.yaml) | [weights](https://download.mindspore.cn/toolkits/mindyolo/yolov3/yolov3-darknet53_300e_mAP455-adfb27af.ckpt) |
| Model Name | Cards | Batch Size | ImageSize | jit_level | graph compile | Box mAP (%) | Params | Recipe | Download |
|:------:| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
| YOLOv3 | 8 | 16 | 640 | O2 | 5 mins | 45.5 | 61.9M | [yaml](./yolov3.yaml) | [weights](https://download.mindspore.cn/toolkits/mindyolo/yolov3/yolov3-darknet53_300e_mAP455-adfb27af.ckpt) |
</details>

<details open markdown>
<summary><b>performance tested on Ascend 910*(8p)</b></summary>

| Name | Scale | BatchSize | ImageSize | Dataset | Box mAP (%) | ms/step | Params | Recipe | Download |
|--------| :---: | :---: | :---: |--------------| :---: | :---: | :---: | :---: | :---: |
| YOLOv3 | Darknet53 | 16 * 8 | 640 | MS COCO 2017 | 46.6 | 396.60 | 61.9M | [yaml](./yolov3.yaml) | [weights](https://download-mindspore.osinfra.cn/toolkits/mindyolo/yolov3/yolov3-darknet53_300e_mAP455-81895f09-910v2.ckpt) |
| Model Name | Cards | Batch Size | ImageSize | jit_level | graph compile | Box mAP (%) | ms/step | Params | Recipe | Download |
|:------:| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
| YOLOv3 | 8 | 16 | 640 | O2 | 5 mins | 46.6 | 396.60 | 61.9M | [yaml](./yolov3.yaml) | [weights](https://download-mindspore.osinfra.cn/toolkits/mindyolo/yolov3/yolov3-darknet53_300e_mAP455-81895f09-910v2.ckpt) |
</details>

<br>
Expand All @@ -38,9 +38,16 @@ We present some updates to YOLO! We made a bunch of little design changes to mak

Please refer to the [GETTING_STARTED](https://github.com/mindspore-lab/mindyolo/blob/master/GETTING_STARTED.md) in MindYOLO for details.

### Requirements

| mindspore | ascend driver | firmware | cann toolkit/kernel
| :-------: | :-----------: | :----------: | :----------------:
| 2.3.1 | 24.1.RC2 | 7.3.0.1.231 | 8.0.RC2.beta1

### Training

<details open>
<details open markdown>
<summary><b>View More</b></summary>

#### - Pretraining Model

Expand Down
7 changes: 7 additions & 0 deletions configs/yolov4/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,9 +52,16 @@ AP (65.7% AP50) for the MS COCO dataset at a realtime speed of 65 FPS on Tesla V

Please refer to the [GETTING_STARTED](https://github.com/mindspore-lab/mindyolo/blob/master/GETTING_STARTED.md) in MindYOLO for details.

### Requirements

| mindspore | ascend driver | firmware | cann toolkit/kernel
| :-------: | :-----------: | :----------: | :----------------:
| 2.3.1 | 24.1.RC2 | 7.3.0.1.231 | 8.0.RC2.beta1

### Training

<details open>
<summary><b>View More</b></summary>

#### - Pretraining Model

Expand Down
7 changes: 7 additions & 0 deletions configs/yolov5/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,16 @@ YOLOv5 is a family of object detection architectures and models pretrained on th

Please refer to the [GETTING_STARTED](https://github.com/mindspore-lab/mindyolo/blob/master/GETTING_STARTED.md) in MindYOLO for details.

### Requirements

| mindspore | ascend driver | firmware | cann toolkit/kernel
| :-------: | :-----------: | :----------: | :----------------:
| 2.3.1 | 24.1.RC2 | 7.3.0.1.231 | 8.0.RC2.beta1

### Training

<details open>
<summary><b>View More</b></summary>

#### - Distributed Training

Expand Down
7 changes: 7 additions & 0 deletions configs/yolov7/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,9 +42,16 @@ YOLOv7 surpasses all known object detectors in both speed and accuracy in the ra

Please refer to the [GETTING_STARTED](https://github.com/mindspore-lab/mindyolo/blob/master/GETTING_STARTED.md) in MindYOLO for details.

### Requirements

| mindspore | ascend driver | firmware | cann toolkit/kernel
| :-------: | :-----------: | :----------: | :----------------:
| 2.3.1 | 24.1.RC2 | 7.3.0.1.231 | 8.0.RC2.beta1

### Training

<details open>
<summary><b>View More</b></summary>

#### - Distributed Training

Expand Down
7 changes: 7 additions & 0 deletions configs/yolov8/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,9 +51,16 @@ Ultralytics YOLOv8, developed by Ultralytics, is a cutting-edge, state-of-the-ar

Please refer to the [GETTING_STARTED](https://github.com/mindspore-lab/mindyolo/blob/master/GETTING_STARTED.md) in MindYOLO for details.

### Requirements

| mindspore | ascend driver | firmware | cann toolkit/kernel
| :-------: | :-----------: | :----------: | :----------------:
| 2.3.1 | 24.1.RC2 | 7.3.0.1.231 | 8.0.RC2.beta1

### Training

<details open>
<summary><b>View More</b></summary>

#### - Distributed Training

Expand Down
7 changes: 7 additions & 0 deletions configs/yolox/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,16 @@ YOLOX is a new high-performance detector with some experienced improvements to Y

Please refer to the [GETTING_STARTED](https://github.com/mindspore-lab/mindyolo/blob/master/GETTING_STARTED.md) in MindYOLO for details.

### Requirements

| mindspore | ascend driver | firmware | cann toolkit/kernel
| :-------: | :-----------: | :----------: | :----------------:
| 2.3.1 | 24.1.RC2 | 7.3.0.1.231 | 8.0.RC2.beta1

### Training

<details open>
<summary><b>View More</b></summary>

#### - Distributed Training

Expand Down
2 changes: 1 addition & 1 deletion docs/en/index.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ The following is the corresponding `mindyolo` versions and supported `mindspore`
| mindyolo | mindspore |
| :--: | :--: |
| master | master |
| 0.4 | 2.3.0 |
| 0.4 | 2.3.1/2.3.0 |
| 0.3 | 2.2.10 |
| 0.2 | 2.0 |
| 0.1 | 1.8 |
Expand Down
2 changes: 1 addition & 1 deletion docs/zh/index.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ MindYOLO基于mindspore实现了最新的YOLO系列算法。以下是mindyolo的
| mindyolo | mindspore |
| :--: | :--: |
| master | master |
| 0.4 | 2.3.0 |
| 0.4 | 2.3.1/2.3.0 |
| 0.3 | 2.2.10 |
| 0.2 | 2.0 |
| 0.1 | 1.8 |
Expand Down

0 comments on commit 687d0ae

Please sign in to comment.