遙感影像語義分割:DinkNet精度提升Trick(更新中)
- 一、資料增強
- 二、Backbone
-
- 編碼器使用SE_ResNet
- 編碼器使用CSPDarkNet
- 三、損失函數
- 四、優化器(待更新)
-
- Adam
- SGD
- 五、學習率更新政策
- 六、總結
一、資料增強
資料增強部分使用的線上資料增強方式,使用的庫是albumentations,訓練過程中使用了翻轉、旋轉、色彩變換、縮放等操作。
import albumentations as albu
def get_training_augmentaion()