前言
相比与torchvision 的 transform 库,albumentations 库提供了更丰富的增强操作和更高的性能;albumentations 通过使用 OpenCV 库实现增强操作,它采用了基于 NumPy 数组的处理方式,并通过并行化和内存优化来提高性能。相比之下,torchvision 的 transform 库是基于 PyTorch 张量的操作,运行在 GPU 上
代码如下(示例):
import albumentations as A
import cv2
import randomdef aug_ok(img, mask, p_value=1, min_value=