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,