transforms.centercrop的用法
时间: 2023-04-26 11:01:58 浏览: 213
Pytoch之torchvision.transforms图像变换实例
transforms.centercrop是一个PyTorch中的图像变换函数,用于将图像居中裁剪为指定大小。
具体使用方法如下:
1. 导入transforms模块:`from torchvision import transforms`
2. 创建CenterCrop对象:`center_crop = transforms.CenterCrop(size)`
其中,size为裁剪后的图像大小,可以是一个整数或一个元组,如(224, 224)。
3. 对图像进行变换:`img = center_crop(img)`
其中,img为待变换的图像。
通过使用transforms.centercrop,可以方便地对图像进行居中裁剪,以便于后续的图像处理和分析。
阅读全文