Skip to content

关于训练PST900数据集? #7

Open

Description

@BingY998

老师,您好!我也在复现您的相关工作,当我尝试训练及测试PST900数据集时总是出现一些错误。可能是相关参数设置的不正确。您的readme文档里默认只提供了MFNet的config文件和train.py,如何训练PST900能否说明或者开源一下呢?万分感谢!

Activity

  1. MathLee commented on Feb 19, 2024

    @MathLee
    Owner

    真是不好意思,由于这个工作完成的较早,中间也更换过服务器,相关代码遗失了,目前也仅有github上这部分代码了。你可以关注下我们近期在RGB-T semantic seg上的其他工作,如[https://github.com/kw717/SGFNet和https://github.com/YingLv1106/CAINet,都有较为完善的代码。

  2. BingY998 commented on Feb 19, 2024

    @BingY998
    Author

    好的,这两篇论文我也在学习,SGFNet也只是提供了MFNet数据集的训练文件,PST900的设置文件和训练代码也没有提供。另外,还有个具体的问题需要请教您,因为PST900数据集图像尺寸为720*1280,当我训练的时候,会出现tensor维度不匹配的错误,我注意到了LASNet.py文件中如下图中的注释,请问这部分应如何更改才能够适用于PST900数据集呢?希望能够得到您的回复!
    image

  3. MathLee commented on Feb 19, 2024

    @MathLee
    Owner

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions