(3)_albumentations数据增强(分割版)
前置知识
在使用pytorch进行dataload,transform之前,需要了解一些数据的知识,许多人使用不同的接口因为不熟悉犯了一些错误。...在这里对一些常用的OpenCV,PIL,skimage进行了一些总结,以及pytorchvision.transorforms的一些简单使用。...数组,转换后的数组为unit8,0-255范围,图像形状是(H,W,C),读入的顺序是RGB
skimage读取进来的图片是numpy数组,是unit8类型,0-255范围,图像形状是(H,W,C),读入的顺序是...注意2:Tensor的形状是[C,H,W],而cv2,plt,PIL,skimage形状都是[H,W,C]
3.1 H×W×C ——> C×H×W
img_cv2.transpose(2,0,1).shape..., 255]的PIL.Image,转换成形状为[C, H, W],取值范围是[0, 1.0]的torch.FloatTensor;
形状为[H, W, C]的numpy.ndarray,转换成形状为[C