pytorch torchvision.transforms.CenterCrop

应用

以图片中心进行裁剪

import torchivision.transforms
import PIL.Image as Image
import torchvision.transforms
 
 
#读入图片
image=Image.open("./test.png")
crop_obj = torchvision.transforms.CenterCrop((224, 224))  #生成一个CenterCrop类的对象,用来将图片从中心裁剪成224*224
image = crop_obj(image)  
image.save("./test2.png", format='PNG')

api

CLASS torchvision.transforms.CenterCrop(size)

参数	描述
size (sequence or int)	输出图片的size(h,w)

参考：
https://pytorch.org/docs/stable/torchvision/transforms.html#torchvision.transforms.CenterCrop