Skip to content

Commit 06a027f

Browse files
author
wangjianfeng
committed
chore(detection): update README.md
1 parent 65aa210 commit 06a027f

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

official/vision/detection/README.md

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -2,23 +2,22 @@
22

33
## 介绍
44

5-
本目录包含了采用MegEngine实现的经典[RetinaNet](https://arxiv.org/pdf/1708.02002>)网络结构,
6-
同时提供了在COCO2017数据集上的完整训练和测试代码。
5+
本目录包含了采用MegEngine实现的经典[RetinaNet](https://arxiv.org/pdf/1708.02002>)网络结构,同时提供了在COCO2017数据集上的完整训练和测试代码。
76

87
网络的性能在COCO2017验证集上的测试结果如下:
98

10-
| 模型 | mAP<br>@5-95 | batch<br>/gpu | gpu | speed<br>(8gpu) | speed<br>(1gpu)|
11-
| --- | --- | --- | --- | --- | --- |
12-
| retinanet-res50-1x-800size | 36.0 | 2 | 2080 | 2.27(it/s) | 3.7(it/s) |
9+
| 模型 | mAP<br>@5-95 | batch<br>/gpu | gpu | speed<br>(8gpu) | speed<br>(1gpu) |
10+
| --- | --- | --- | --- | --- | --- |
11+
| retinanet-res50-coco-1x-800size | 36.0 | 2 | 2080ti | 2.27(it/s) | 3.7(it/s) |
1312

14-
* MegEngine v0.3.0
13+
* MegEngine v0.4.0
1514

1615
## 如何使用
1716

1817
模型训练好之后,可以通过如下命令测试单张图片:
1918

2019
```bash
21-
python3 tools/inference.py -f retinanet_res50_1x_800size.py \
20+
python3 tools/inference.py -f retinanet_res50_coco_1x_800size.py \
2221
-i ../../assets/cat.jpg \
2322
-m /path/to/retinanet_weights.pkl
2423
```
@@ -35,8 +34,8 @@ python3 tools/inference.py -f retinanet_res50_1x_800size.py \
3534

3635
## 如何训练
3736

38-
1. 在开始训练前,请确保已经下载解压好[COCO数据集](http://cocodataset.org/#download)
39-
并放在合适的数据目录下,准备好的数据集的目录结构如下所示(目前默认使用coco2017的数据集):
37+
1. 在开始训练前,请确保已经下载解压好[COCO2017数据集](http://cocodataset.org/#download)
38+
并放在合适的数据目录下,准备好的数据集的目录结构如下所示(目前默认使用COCO2017数据集):
4039

4140
```
4241
/path/to/
@@ -46,14 +45,14 @@ python3 tools/inference.py -f retinanet_res50_1x_800size.py \
4645
| |val2017
4746
```
4847

49-
2. 准备预训练的`backbone`网络权重:可使用 megengine.hub 下载`megengine`官方提供的在ImageNet上训练的resnet50模型, 并存放在 `/path/to/pretrain.pkl`
48+
2. 准备预训练的`backbone`网络权重:可使用 megengine.hub 下载`megengine`官方提供的在ImageNet上训练的ResNet-50模型, 并存放在 `/path/to/pretrain.pkl`
5049

5150
3. 在开始运行本目录下的代码之前,请确保按照[README](../../../README.md)进行了正确的环境配置。
5251

5352
4. 开始训练:
5453

5554
```bash
56-
python3 tools/train.py -f retinanet_res50_1x_800size.py \
55+
python3 tools/train.py -f retinanet_res50_coco_1x_800size.py \
5756
-n 8 \
5857
--batch_size 2 \
5958
-w /path/to/pretrain.pkl
@@ -65,7 +64,7 @@ python3 tools/train.py -f retinanet_res50_1x_800size.py \
6564
- `-n`, 用于训练的devices(gpu)数量,默认使用所有可用的gpu.
6665
- `-w`, 预训练的backbone网络权重的路径。
6766
- `--batch_size`,训练时采用的`batch size`, 默认2,表示每张卡训2张图。
68-
- `--dataset-dir`, coco数据集的根目录,默认`/data/datasets/coco`
67+
- `--dataset-dir`, COCO2017数据集的上级目录,默认`/data/datasets`
6968

7069
默认情况下模型会存在 `log-of-retinanet_res50_1x_800size`目录下。
7170

@@ -74,18 +73,18 @@ python3 tools/train.py -f retinanet_res50_1x_800size.py \
7473
在训练的过程中,可以通过如下命令测试模型在`COCO2017`验证集的性能:
7574

7675
```bash
77-
python3 tools/test.py -n 8 \
78-
-f retinanet_res50_1x_800size.py \
76+
python3 tools/test.py -f retinanet_res50_coco_1x_800size.py \
77+
-n 8 \
7978
--model /path/to/retinanet_weights.pt \
80-
--dataset_dir /data/datasets/coco
79+
--dataset_dir /data/datasets
8180
```
8281

8382
`tools/test.py`的命令行选项如下:
8483

8584
- `-f`, 所需要测试的网络结构描述文件。
8685
- `-n`, 用于测试的devices(gpu)数量,默认1;
8786
- `--model`, 需要测试的模型;可以从顶部的表格中下载训练好的检测器权重, 也可以用自行训练好的权重。
88-
- `--dataset_dir`coco数据集的根目录,默认`/data/datasets`
87+
- `--dataset_dir`COCO2017数据集的上级目录,默认`/data/datasets`
8988

9089
## 参考文献
9190

0 commit comments

Comments
 (0)