@@ -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```
6565import paddle
6666print(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+
95103GPU单卡训练
96104``` bash
97105export CUDA_VISIBLE_DEVICES=0
@@ -100,21 +108,21 @@ python3.7 tools/train.py -c configs/dota/s2anet_1x_spine.yml
100108
101109GPU多卡训练
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