Pytorch函数torchvision.transforms.

由于需要在深度学习网络中加一层裁剪层，所以在torchvision.transforms.functional.crop的基础上继承torch.nn.Module类自定义裁剪层。

自定义裁剪层

在pytorch1.7下能正常运行包含裁剪层的网络，切换到pytorch1.6下后报错

错误信息

发现1.6版本下该函数只接受PIL输入

pytorch1.6下torchvision.transforms.functional.crop

但pytorch1.7版本支持裁剪tensor

pytorch1.7下torchvision.transforms.functional.crop

解决方法：升级pytorch，或添加数据类型转换程序