由于需要在深度學習網(wǎng)絡中加一層裁剪層芹关,所以在torchvision.transforms.functional.crop的基礎上繼承torch.nn.Module類自定義裁剪層续挟。
自定義裁剪層
在pytorch1.7下能正常運行包含裁剪層的網(wǎng)絡,切換到pytorch1.6下后報錯
錯誤信息
發(fā)現(xiàn)1.6版本下該函數(shù)只接受PIL輸入
pytorch1.6下torchvision.transforms.functional.crop
但pytorch1.7版本支持裁剪tensor
pytorch1.7下torchvision.transforms.functional.crop
解決方法:升級pytorch侥衬,或添加數(shù)據(jù)類型轉換程序