search

Joazs / Infrared-Small-Target-Segmentation-with-GNN

IRGraphSeg: Infrared Small Target Detection Based on Hierarchical GNN
5 stars 0 forks source link

open source code #1

Open Web-AK opened 6 months ago

Web-AK commented 6 months ago

May I ask when the author will open source all the code?

Joazs commented 6 months ago

感谢您的关注，2024年5月1日前将开源所有代码。

Joazs commented 5 months ago

May I ask when the author will open source all the code?

您好，代码已开源。如果遇到什么问题，可以问我。感谢您的关注。

Web-AK commented 4 months ago

尊敬的作者，您好。 我在运行代码的过程中遇到一个问题，batch-size设置成4.运行会报错：

image image 
    将batch-size修改为1或者2就不会报这个错误，想请教请教。
Joazs commented 4 months ago

我感觉是超显存了。这个程序占用显存挺大的。你可以尝试： 将batch_size设置少一点。或者将graph_conv改成mr的。 还有，为啥你显示了2次“如果快乐太难，那我祝你平安”。正常应该是显示一次的，你检查一下代码。

Web-AK commented 4 months ago

嗯嗯 我已收到