阿里天池2019年县域农业大脑AI挑战赛,使用deeplabv3
做卫星图语义分割。
1. pyvips
sudo apt install libvips
pip install pyvips
2. opencv
pip install opencv-python
- 将RGBA图片处理为RGB图片,其中透明度为0的区域像素值设置为
(0, 0, 0)
,其他则直接提取出RGB通道。 - 数据增强,包括多尺度裁剪、随机裁剪、多角度旋转(90/180/270)、镜像、翻转等。
- 减去均值,做归一化。
- 随机高斯模糊、对比度、色调等操作。
Scale | Mean | Std |
---|---|---|
320*320 | [133.40, 133.06, 115.60] |
[60.34, 53.46, 51.97] |
480*480 | [131.62, 131.28, 114.05] |
[61.86, 55. 53.30] |
960*960 | [126.33, 126.01, 109.47] |
[65.89, 59.96, 56.82] |
1280*1280 | [123.90, 123.59, 107.37] |
[67.53, 61.86, 58.25] |
采取pytorch
内置的deeplabv3_resnet101
网络,由于PASCAL数据集与卫星云图数据集相差较远,所以不采用预训练权重。
-
metrics
函数 -
refine
函数
- [0.20] refined, 1280, scales = [480, 600, 960, 1280]
- [0.22] no refine, 1280, scales = [480, 600, 960, 1280]
- [0.22] no refine, 480, scales = [224, 320, 480]
- [0.23] no refine, focal weighted loss, 256, scales = [256, 480]