图像分割数据集的相关操作(二)—— albumentations 数据增强


前言

相比与torchvision 的 transform 库,albumentations 库提供了更丰富的增强操作和更高的性能;albumentations 通过使用 OpenCV 库实现增强操作,它采用了基于 NumPy 数组的处理方式,并通过并行化和内存优化来提高性能。相比之下,torchvision 的 transform 库是基于 PyTorch 张量的操作,运行在 GPU 上

代码如下(示例):

import albumentations as A
import cv2
import random

def aug_ok(img, mask, p_value=1, min_value=

你可能感兴趣的:(AI图像算法,计算机视觉,深度学习)