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


一、albumentations和transfroms的主要区别

(1)torchvision.transforms模块接受PIL图像对象作为输入,albumentations接受NumPy数组作为输入,即

    image = Image.open('C:/Users/02073-0_WA.jpg')
    # image = Image.open('C:/Users/02073-0_WA.jpg').convert('L'

猜你喜欢

转载自blog.csdn.net/qq_43199575/article/details/133910033