老师,您好!我也在复现您的相关工作,当我尝试训练及测试PST900数据集时总是出现一些错误。可能是相关参数设置的不正确。您的readme文档里默认只提供了MFNet的config文件和train.py,如何训练PST900能否说明或者开源一下呢?万分感谢!
Activity
MathLee commentedon Feb 19, 2024
真是不好意思,由于这个工作完成的较早,中间也更换过服务器,相关代码遗失了,目前也仅有github上这部分代码了。你可以关注下我们近期在RGB-T semantic seg上的其他工作,如[https://github.com/kw717/SGFNet和https://github.com/YingLv1106/CAINet,都有较为完善的代码。
BingY998 commentedon Feb 19, 2024
好的,这两篇论文我也在学习,SGFNet也只是提供了MFNet数据集的训练文件,PST900的设置文件和训练代码也没有提供。另外,还有个具体的问题需要请教您,因为PST900数据集图像尺寸为720*1280,当我训练的时候,会出现tensor维度不匹配的错误,我注意到了LASNet.py文件中如下图中的注释,请问这部分应如何更改才能够适用于PST900数据集呢?希望能够得到您的回复!

MathLee commentedon Feb 19, 2024
SGFNet代码相关的问题可以发邮件给一作,并抄送给我和Prof. Liu。对于PST900数据集,将LASNet.py文件中line 157的2倍上采样注释掉,然后不注释line 195的代码,其中line 195代码的fea_size是x4的fea_size。如此,应该可以跑通。