BuildingExtraction
BuildingExtraction copied to clipboard
KyanChen
Paper: Building Extraction from Remote Sensing Images with Sparse Token Transformers
img,label 0,C:\Users\Nine\Desktop\BuildingExtraction-main\Data\test_samples\img\219561_112236_1.png,C:\Users\Nine\Desktop\BuildingExtraction-main\Data\test_samples\label\austin2_01179_258_289.png
我没对main.py进行任何改动，训练完一个epoch的时候，发生了如题错误。我使用的是32g内存，24g显存3090的电脑进行训练，请问我我该如何修改防止爆内存？
数据集问题
感谢分享代码！有个关于数据集的问题想问一下。 刚刚接触建筑物提取，文中提到的INRIA数据集，请问用于实验的文件夹格式是什么样子的啊，我下载的数据集包含test和train两个文件夹，只在train文件夹里有gt文件夹，请问是将Cutimgsegwithlabel中的parent_path直接替换成上述我说的train文件夹的路径就可以了吗？如果按照这个路径来，得不到分割结果欸，麻烦大佬给解答下，万分感谢！
在复现您的代码时，分别用您已经训练好的inria和whu模型去测试数据，发现在Inria数据集上可以取得很好的提取效果，但是在whu数据集上发现效果比较一般。另外，我在对国内其他城市进行建筑物提取的时候发现效果比较一般，请问您有什么建议嘛，谢谢！
输入图像大小
作者您好，再次打扰您了，请问STT是只允许输入512x512的吗？我刚才试了试256x256了的，提示我卷积核大小不对，是否有办法实现256x256的呢？
您好！我用模型训练了自己的数据集，但是我没太搞明白Result->test文件下面的图片产生的顺序，而且为什么没有生成所有在Data->test_samples的图片测试结果。或者说有没有测试文件呀，不然没有办法计算iou_score，无法查看训练效果。谢谢！
