遥感影像语义分割:DinkNet精度提升Trick(更新中)
- 一、数据增强
- 二、Backbone
-
- 编码器使用SE_ResNet
- 编码器使用CSPDarkNet
- 三、损失函数
- 四、优化器(待更新)
-
- Adam
- SGD
- 五、学习率更新策略
- 六、总结
一、数据增强
数据增强部分使用的在线数据增强方式,使用的库是albumentations,训练过程中使用了翻转、旋转、色彩变换、缩放等操作。
import albumentations as albu
def get_training_augmentaion()