前言
Pytorch的transforms模块含有的裁剪方式,通常是按照设置的尺寸大小和长宽比来进行随机裁剪。在实际项目中,往往需要有选择性地进行裁剪。在本文中,记录下项目中使用到的一些自定义随机方式和自己的一些思考(可能是错的),供日后参考
一、单边裁剪
情形:如果遇到一些长条形的特殊尺寸,如图片尺寸为(H=480,W=32),而对此只想要裁剪较长的那一边,较短的一边保持不变,则需要自己定义裁剪操作
代码如下:
def random_crop_image(image, mask,crop_size, aspect_ratio_range=(0.5,