问题:高光谱图像输入为5维,即(batch_size, channels, bands, height, width) ,需要将其分为多个大小为(k, k, k)的小块,但pytorch的unfold函数只支持输入是四维的情况
解决方法:使用单独的包
https://github.com/f-dangel/unfoldNd
版权声明:本文为pp2373886592原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
问题:高光谱图像输入为5维,即(batch_size, channels, bands, height, width) ,需要将其分为多个大小为(k, k, k)的小块,但pytorch的unfold函数只支持输入是四维的情况
解决方法:使用单独的包
https://github.com/f-dangel/unfoldNd