Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
f5dacedd
P
PaddleDetection
项目概览
s920243400
/
PaddleDetection
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleDetection
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
f5dacedd
编写于
8月 17, 2021
作者:
C
cnn
提交者:
GitHub
8月 17, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update doc, test=document_fix (#3991)
上级
6734c2e1
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
16 addition
and
6 deletion
+16
-6
configs/dota/README.md
configs/dota/README.md
+14
-4
configs/dota/s2anet_alignconv_2x_dota.yml
configs/dota/s2anet_alignconv_2x_dota.yml
+2
-2
未找到文件。
configs/dota/README.md
浏览文件 @
f5dacedd
...
...
@@ -109,13 +109,23 @@ python3.7 -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c
### 3. 评估
```
bash
python3.7 tools/eval.py
-c
configs/dota/s2anet_1x_spine.yml
-o
weights
=
output/s2anet_1x_spine/model_final.pdparams
# 使用提供训练好的模型评估
python3.7 tools/eval.py
-c
configs/dota/s2anet_1x_spine.yml
-o
weights
=
https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams
```
** 注意:**
dota数据集中是train和val数据作为训练集一起训练的,对dota数据集进行评估时需要自定义设置评估数据集配置。
** 注意:**
(1) dota数据集中是train和val数据作为训练集一起训练的,对dota数据集进行评估时需要自定义设置评估数据集配置。
(2) 骨骼数据集是由分割数据转换而来,由于椎间盘不同类别对于检测任务而言区别很小,且s2anet算法最后得出的分数较低,评估时默认阈值为0.5,mAP较低是正常的。建议通过可视化查看检测结果。
### 4. 预测
执行如下命令,会将图像预测结果保存到
`output`
文件夹下。
```
bash
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
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
--draw_threshold
=
0.3
```
使用提供训练好的模型预测:
```
bash
python3.7 tools/infer.py
-c
configs/dota/s2anet_1x_spine.yml
-o
weights
=
https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams
--infer_img
=
demo/39006.jpg
--draw_threshold
=
0.3
```
### 5. DOTA数据评估
...
...
@@ -124,7 +134,7 @@ python3.7 tools/infer.py -c configs/dota/s2anet_1x_spine.yml -o weights=output/s
python3.7 tools/infer.py -c configs/dota/s2anet_alignconv_2x_dota.yml -o weights=./weights/s2anet_alignconv_2x_dota.pdparams --infer_dir=dota_test_images --draw_threshold=0.05 --save_txt=True --output_dir=output
```
请参考
[
DOTA_devkit
](
https://github.com/CAPTAIN-WHU/DOTA_devkit
)
生成评估文件,评估文件格式请参考
[
DOTA Test
](
http://captain.whu.edu.cn/DOTAweb/tasks.html
)
,生成zip文件,每个类一个txt文件,txt文件中每行格式为:
`image_id score x1 y1 x2 y2 x3 y3 x4 y4`
,提交服务器进行评估。
请参考
[
DOTA_devkit
](
https://github.com/CAPTAIN-WHU/DOTA_devkit
)
生成评估文件,评估文件格式请参考
[
DOTA Test
](
http://captain.whu.edu.cn/DOTAweb/tasks.html
)
,生成zip文件,每个类一个txt文件,txt文件中每行格式为:
`image_id score x1 y1 x2 y2 x3 y3 x4 y4`
,提交服务器进行评估。
您也可以参考
`dataset/dota_coco/dota_generate_test_result.py`
脚本生成评估文件,提交到服务器。
## 模型库
...
...
@@ -132,7 +142,7 @@ python3.7 tools/infer.py -c configs/dota/s2anet_alignconv_2x_dota.yml -o weights
| 模型 | Conv类型 | mAP | 模型下载 | 配置文件 |
|:-----------:|:----------:|:--------:| :----------:| :---------: |
| 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_1
x_dota.yml
)
|
| S2ANet | Conv | 71.42 |
[
model
](
https://paddledet.bj.bcebos.com/models/s2anet_conv_
2x_dota.pdparams
)
|
[
config
](
https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/dota/s2anet_conv_2
x_dota.yml
)
|
| 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
)
|
**注意:**
这里使用
`multiclass_nms`
,与原作者使用nms略有不同。
...
...
configs/dota/s2anet_alignconv_2x_dota.yml
浏览文件 @
f5dacedd
...
...
@@ -21,7 +21,7 @@ S2ANetHead:
align_conv_type
:
'
AlignConv'
# AlignConv Conv
align_conv_size
:
3
use_sigmoid_cls
:
True
reg_loss_weight
:
[
1.0
,
1.0
,
1.0
,
1.0
,
1.
05
]
cls_loss_weight
:
[
1.
05
,
1.0
]
reg_loss_weight
:
[
1.0
,
1.0
,
1.0
,
1.0
,
1.
1
]
cls_loss_weight
:
[
1.
1
,
1.05
]
reg_loss_type
:
'
l1'
is_training
:
True
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录