search

ViTAE-Transformer / SAMRS

The official repo for [NeurIPS'23] "SAMRS: Scaling-up Remote Sensing Segmentation Dataset with Segment Anything Model"
288 stars 14 forks source link

作者您好，请问vib那个模型的泛化性怎么样？如果用您的数据集训练后用到地物分割这种场景中精度会是怎样的？ #18

Open funny000 opened 7 months ago

DotWang commented 7 months ago

@funny000 可以参考我们最新的文章：MTP: Advancing Remote Sensing Foundation Model via Multi-Task Pretraining

funny000 commented 7 months ago

好的，谢谢作者

funny000 commented 7 months ago

@DotWang 作者好，还有个问题，在配置微调环境的时候，用新版本的torch2.1.1，cuda12.2，在编译dcnv3时候报错了，目前在配置仓库指定的环境，除了指定的环境，还兼容其他版本的环境吗，目前cuda环境是12.2，

DotWang commented 7 months ago

@funny000 去掉子文件夹中所有扩展名是hip的，然后重新编译

关于安装，可参考：https://github.com/ViTAE-Transformer/MTP/issues/2

funny000 commented 7 months ago

谢谢作者，环境配置好了，正尝试微调模型。