pytorch 实现数据增强分类 albumentations的使用

摘要 albumentations包是一种针对数据增强专门写的API，里面基本包含大量的数据增强手段，比起pytorch自带的ttransform更丰富，搭配使用效果更好。 代码和效果 import albumentations import cv2 from PIL import Image, ImageDraw import numpy as np from albumentations import (Blur,Flip,ShiftScaleRotate,GridDistortion,ElasticTransform,HorizontalFlip,CenterCrop,