天天看點

遙感影像語義分割:資料增強一、albumentations介紹

遙感影像語義分割:資料增強

  • 一、albumentations介紹
  • 二、在項目中的使用方式
遙感影像語義分割:資料增強一、albumentations介紹

深度學習訓練的模型需要大量的資料集作為支撐才能有比較好的泛化能力,但是實際項目中由于資料資源和标注難度的限制而難以獲得的足夠多的資料,這時候我們可以通過資料增強的方法來擴充我們的資料集。下面介紹一個我在做遙感影像深度學習時常用的資料增強庫albumentations。

一、albumentations介紹

albumentations是一個用來做圖像增強的Python開源庫,使用起來非常簡單友善,主要包括像素上的變換和空間上的變換。

1.1、基本使用方法

import albumentations as A
import cv2

           

繼續閱讀