xdFai / SCTransNet

[IEEE TGRS 2024] SCTransNet: Spatial-channel Cross Transformer Network for Infrared Small Target Detection
88 stars 1 forks source link

我在单个NUAA-SIRST和 NUDT-SIRST数据集上训练会出现这样的报错 #19

Open push-xiting opened 3 hours ago

push-xiting commented 3 hours ago

d0dc337913701cb8af9e97c88800c3df 这是我的NUAA-SIRST的数据集结构 image image

还有我想知道怎么三个数据集整合为SIRST3

xdFai commented 2 hours ago

同学你好。 第一个问题是因为路径不对(找不到png才读取的bmp),建议你debug,然后定位到dataset函数内部,查看下真实读取的数据路径。 第二个问题将三个数据集的 images 以及masks放到一起,同理 img_idx放到用一个txt文件。