Skip to content

Commit d6c8947

Browse files
author
cnn
authored
fix doc, test=document_fix (PaddlePaddle#3862)
1 parent cc266e5 commit d6c8947

File tree

1 file changed

+17
-9
lines changed

1 file changed

+17
-9
lines changed

configs/dota/README.md

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -50,17 +50,17 @@ DOTA数据集中总共有2806张图像,其中1411张图像作为训练集,45
5050
旋转框IOU计算OP[ext_op](../../ppdet/ext_op)是参考Paddle[自定义外部算子](https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/07_new_op/new_custom_op.html) 的方式开发。
5151

5252
若使用旋转框IOU计算OP,需要环境满足:
53-
- PaddlePaddle >= 2.0.1
53+
- PaddlePaddle >= 2.1.1
5454
- GCC == 8.2
5555

56-
推荐使用docker镜像[paddle:2.0.1-gpu-cuda10.1-cudnn7](registry.baidubce.com/paddlepaddle/paddle:2.0.1-gpu-cuda10.1-cudnn7)
56+
推荐使用docker镜像[paddle:2.0.1-gpu-cuda10.1-cudnn7](registry.baidubce.com/paddlepaddle/paddle:2.1.1-gpu-cuda10.2-cudnn7)
5757

5858
执行如下命令下载镜像并启动容器:
5959
```
60-
sudo nvidia-docker run -it --name paddle_s2anet -v $PWD:/paddle --network=host registry.baidubce.com/paddlepaddle/paddle:2.0.1-gpu-cuda10.1-cudnn7 /bin/bash
60+
sudo nvidia-docker run -it --name paddle_s2anet -v $PWD:/paddle --network=host registry.baidubce.com/paddlepaddle/paddle:2.1.1-gpu-cuda10.2-cudnn7 /bin/bash
6161
```
6262

63-
镜像中paddle2.0.1已安装好,进入python3.7,执行如下代码检查paddle安装是否正常:
63+
镜像中paddle已安装好,进入python3.7,执行如下代码检查paddle安装是否正常:
6464
```
6565
import paddle
6666
print(paddle.__version__)
@@ -92,6 +92,14 @@ python3.7 test.py
9292
**注意:**
9393
配置文件中学习率是按照4卡GPU训练设置的,如果使用单卡GPU训练,请将学习率设置为原来的1/4。
9494

95+
准备数据
96+
```bash
97+
cd dataset/spine_coco
98+
wget https://paddledet.bj.bcebos.com/data/spine_coco.tar
99+
tar -xvf spine_coco.tar
100+
cd ../../
101+
```
102+
95103
GPU单卡训练
96104
```bash
97105
export CUDA_VISIBLE_DEVICES=0
@@ -100,21 +108,21 @@ python3.7 tools/train.py -c configs/dota/s2anet_1x_spine.yml
100108

101109
GPU多卡训练
102110
```bash
103-
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
104-
python3.7 -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/dota/s2anet_1x_spine.yml
111+
export CUDA_VISIBLE_DEVICES=0,1,2,3
112+
python3.7 -m paddle.distributed.launch --gpus 0,1,2,3 tools/train.py -c configs/dota/s2anet_1x_spine.yml
105113
```
106114

107115
可以通过`--eval`开启边训练边测试。
108116

109117
### 2. 评估
110118
```bash
111-
python3.7 tools/eval.py -c configs/dota/s2anet_1x_spine.yml -o weitghts=output/s2anet_1x_spine/model_final.pdparams
119+
python3.7 tools/eval.py -c configs/dota/s2anet_1x_spine.yml -o weights=output/s2anet_1x_spine/model_final.pdparams
112120
```
113121

114122
### 3. 预测
115123
执行如下命令,会将图像预测结果保存到`output_dir`文件夹下。
116124
```bash
117-
python3.7 tools/infer.py -c configs/dota/s2anet_1x_spine.yml -o weitghts=output/s2anet_1x_spine/model_final.pdparams --infer_img=demo/39006.jpg
125+
python3.7 tools/infer.py -c configs/dota/s2anet_1x_spine.yml -o weights=output/s2anet_1x_spine/model_final.pdparams --infer_img=demo/39006.jpg
118126
```
119127

120128
### 4. DOTA数据评估
@@ -134,7 +142,7 @@ python3.7 tools/infer.py -c configs/dota/s2anet_1x_dota.yml -o weights=./weights
134142
| S2ANet | Conv | 71.42 | [model](https://paddledet.bj.bcebos.com/models/s2anet_conv_1x_dota.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/dota/s2anet_conv_1x_dota.yml) |
135143
| S2ANet | AlignConv | 74.0 | [model](https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/dota/s2anet_alignconv_2x_dota.yml) |
136144

137-
**注意:**这里使用`multiclass_nms`,与原作者使用nms略有不同,精度相比原始论文中高0.15 (71.27-->71.42)
145+
**注意:** 这里使用`multiclass_nms`,与原作者使用nms略有不同。
138146

139147

140148
## 预测部署

0 commit comments

Comments
 (0)