PaddleSeg动态图遥感比赛Baseline
收藏
快速回复
常规赛:遥感影像地块分割 文章学习资料 6391 32
PaddleSeg动态图遥感比赛Baseline
收藏
快速回复
常规赛:遥感影像地块分割 文章学习资料 6391 32

2020 CCF BDCI 遥感影像地块分割比赛主办方提供了PaddleSeg静态图版本作为baseline,现正值飞桨框架推荐使用动态图,作为飞桨的图像分割工具套件PaddleSeg动态图版也已经适配了Paddle 2.0rc版。Seg的动态图版本是今后推荐使用的,第一期支持的模型数和精度也已经超过静态图版了,但目前在处于实验阶段,并未全面向用户推出,借着这次比赛机会,我们也测试了动态图的性能。

作为动态图的baseline,我们选择了一个比较简单的分割模型,HRNet_W48+FCN,并未做任何调参,配置文件以及测试结果在这里能找到:

PaddleSeg动态图主目录:https://github.com/PaddlePaddle/PaddleSeg/tree/develop/dygraph

遥感分割比赛配置文件:https://github.com/PaddlePaddle/PaddleSeg/tree/develop/dygraph/contrib/remote_sensing

数据准备参考:https://aistudio.baidu.com/aistudio/projectdetail/1090790

训练方式:

python -m paddle.distributed.launch train.py --config contrib/remote_sensing/fcn_hrnetw48_ccf_256x256_160k.yml --save_dir save_model_path --save_interval 8000 --log_iters 200 --do_eval --use_vdl

单尺度评估:

python -m paddle.distributed.launch val.py --config contrib/remote_sensing/fcn_hrnetw48_ccf_256x256_160k.yml --model_path save_model_path/best_model/model.pdparams

多尺度评估:

python -m paddle.distributed.launch val.py --config contrib/remote_sensing/fcn_hrnetw48_ccf_256x256_160k.yml --model_path save_model_path/best_model/model.pdparams --aug_eval --scales 1.0 1.25  --flip_horizontal

我们采用了1.0和1.25倍的多尺度做一个示例(同学们可以尝试更多组合),另外加了水平翻转。

生成上传图片:

python predict.py --config contrib/remote_sensing/fcn_hrnetw48_ccf_256x256_160k.yml --model_path save_model_path/best_model/model.pdparams --image_path dataset/rs_data/img_testA --aug_pred --scales 1.0 1.25 --flip_horizontal --save_dir output_images_path

和多尺度评估方式类似。

我们最后得到的结果:在单尺度验证集mIoU是71.55%,增加多尺寸和翻转验证集mIoU是72.42%,最后在初赛的测试集mIoU是70.36%,结果和训练模型都能在PaddleSeg动态图里面找到:

https://github.com/PaddlePaddle/PaddleSeg/tree/develop/dygraph/contrib/remote_sensing

目前PaddleSeg动态图版本正在快马加鞭的完善文档中,可能有些遗漏,有不清楚的欢迎直接加QQ群(850378321)圈我们,如果对大家有帮助,希望能在github上给我们PaddleSeg点小星星。

https://github.com/PaddlePaddle/PaddleSeg

10
收藏
回复
全部评论(32)
时间顺序
Seigato
#22 回复于2020-11
Irene2712 #20
请问AI STUIDIO平台的GPU跑完测试需要多久呢

我测试了一下,不加多尺度需要约18分钟,加了多尺度大概需要1小时10分钟

0
回复
Irene2712
#23 回复于2020-11
Seigato #22
我测试了一下,不加多尺度需要约18分钟,加了多尺度大概需要1小时10分钟

单卡v100训练了多久呀

0
回复
略略略1111110
#24 回复于2020-11

快结束的时候出这么一个高分baseline,这是让各个团队之前付出的努力打水漂?你们PaddleSeg团队怎么做事的?真想骂人

2
回复
xxxue16
#25 回复于2020-11

请问为什么用了develop分支还是生成jpg图片啊

0
回复
z
zpx5511
#26 回复于2020-11

module 'paddle.optimizer' has no attribute 'lr' 

怎么回事

0
回复
Elegantgod
#27 回复于2020-11

验证集的评估太慢慢慢慢慢了,我拿十分之一的数据1w多点图片评估一次要二十多分钟

0
回复
m
michaelowenliu
#28 回复于2020-12
zpx5511 #26
module 'paddle.optimizer' has no attribute 'lr'  怎么回事

版本不对,请安装paddlepaddle 2.0rc版本

0
回复
m
michaelowenliu
#29 回复于2020-12
验证集的评估太慢慢慢慢慢了,我拿十分之一的数据1w多点图片评估一次要二十多分钟

图片比较多再加多尺度翻转的话确实比较慢,可以多GPU评估加速

0
回复
m
michaelowenliu
#30 回复于2020-12
zpx5511 #26
module 'paddle.optimizer' has no attribute 'lr'  怎么回事

现在的版本没问题的

0
回复
大笨熊olo
#31 回复于2020-12

后期还会有遥感分割方面的比赛吗

 

0
回复
大笨熊olo
#32 回复于2020-12

希望能出那种在aistudio上面单卡能跑完的比赛,算力卡是痛点

0
回复
Tos
#33 回复于2021-01

非常好

0
回复
在@后输入用户全名并按空格结束,可艾特全站任一用户