pytorch API:随机裁剪

def imgAugTest(imgDir):
    img=Image.open(imgDir)
    print('the size of raw img:{}'.format(img.size))
    cropImg=transforms.RandomResizedCrop(224)(img)
    print('the size of new img {}'.format(cropImg.size))
    cropImg01=transforms.RandomResizedCrop(224)(img)
    cropImg02=transforms.RandomResizedCrop(224)(img)

    plt.figure()
    plt.subplot(2,2,1),plt.imshow(img),plt.title("raw")
    plt.subplot(2, 2, 2), plt.imshow(cropImg), plt.title("img 00")
    plt.subplot(2, 2, 3), plt.imshow(cropImg01), plt.title("img 01")
    plt.subplot(2, 2, 4), plt.imshow(cropImg02), plt.title("img 02")
    plt.show()

效果图：