天天看点

遥感影像语义分割:数据增强一、albumentations介绍

遥感影像语义分割:数据增强

  • 一、albumentations介绍
  • 二、在项目中的使用方式
遥感影像语义分割:数据增强一、albumentations介绍

深度学习训练的模型需要大量的数据集作为支撑才能有比较好的泛化能力,但是实际项目中由于数据资源和标注难度的限制而难以获得的足够多的数据,这时候我们可以通过数据增强的方法来扩充我们的数据集。下面介绍一个我在做遥感影像深度学习时常用的数据增强库albumentations。

一、albumentations介绍

albumentations是一个用来做图像增强的Python开源库,使用起来非常简单方便,主要包括像素上的变换和空间上的变换。

1.1、基本使用方法

import albumentations as A
import cv2

           

继续阅读