search

TinaLRJ / Multi-frame-infrared-small-target-detection-DTUM

29 stars 3 forks source link

关于在其它数据集上训练的问题 #6

Open YangBo0411 opened 1 week ago

YangBo0411 commented 1 week ago

作者你好，感谢你做出的杰出工作。想请教下，这个项目对数据集的加载方式好像只适用于NUDT-MIRSDT数据集，如果我想使用其它现今比较通用的多帧数据集如DAUB（A dataset for infrared image dim-small aircraft target detection and tracking under ground/air background）、ITSDT（A dataset for infrared time-sensitive target detection and tracking for air-ground application）以及IRDST （Receptive-Field and Direction Induced Attention Network for Infrared Dim Small Target Detection With a Large-Scale Dataset IRDST），应该怎么做？

TinaLRJ commented 1 week ago

已更新了在IRDST数据集的加载，其他数据集类似，可以根据自己的需求更改

YangBo0411 commented 6 days ago

感谢及时回复，已经运行起来了。

TinaLRJ commented 5 days ago

框架对数据集没有选择性，都适用。

YangBo0411 commented 5 days ago

框架对数据集没有选择性，都适用。

我在使用ResUNet_DTUM的时候会出现网络结构层之间输入输出不匹配的问题，使用DNANet_DTUM的时候好像又能跑，我在捣鼓捣鼓试试。

TinaLRJ commented 5 days ago

可能是有的网络要求输入必须是2的几次方的整数倍，所以数据载入时候需要对此进行适配，比如253234图像resize成256256。

TinaLRJ commented 5 days ago

感谢关注，还有更多问题欢迎继续交流（我的邮箱：liruojing@nudt.edu.cn）

YangBo0411 commented 5 days ago

感谢关注，还有更多问题欢迎继续交流（我的邮箱：liruojing@nudt.edu.cn）

好的好的，感谢您的及时解惑。

bbzqr commented 5 days ago

作者您好，感谢您的工作。关于您在IRDST数据集上的工作，我注意到需要加载数据集中的mask标注图。目前我已经下载的DAUB和IRDST数据集中没有mask标注图的部分。您是下载了包含mask标注的IRDST数据集还是您自己生成制作了对应的mask图呢？此外，我尝试通过一些标注方法生成了IRDST和DAUB数据集的mask图，正巧我还没有对您原工作中的数据加载器部分进行调整，请问我能否直接使用您更新的IRDST数据集的加载器进行训练和测试？我是否还需要类似于您之前工作中的mat文件一样，进行一些额外的处理，比如添加部分噪声等。

TinaLRJ commented 4 days ago

IRDST本身提供了mask的，直接下载就可以。对于没有mask图的（如DAUB），可以考虑文中3.5节介绍的点监督方法，根据数据集提供的标注框，自行生成点级mask，结合HPM loss进行训练，效果直逼全监督，可以一试。不过我代码里没有对背景运动的场景进行配准，近期考虑在代码里增加对齐预处理，建议代码更新后在DAUB等背景运动的数据集上训练。

近期更新的IRDST加载器可以直接用，其他数据集也可以根据更新的IRDST加载器进行修改。

我在.mat文件中并没有额外增加噪声，跟.png图像是一样的，只是为了方便同时加载5帧图片，提前把5帧图像合在一起形成了.mat

-----原始邮件----- 发件人: "bbzqr" @.>(由 "noreply" @.> 代发) 发送时间: 2024-10-31 11:58:01 收件人: "TinaLRJ/Multi-frame-infrared-small-target-detection-DTUM" @.> 抄送: "TinaLRJ" @.>,"Comment" @.***> 主题: Re: [TinaLRJ/Multi-frame-infrared-small-target-detection-DTUM] 关于在其它数据集上训练的问题 (Issue #6)

作者您好，感谢您的工作。关于您在IRDST数据集上的工作，我注意到需要加载数据集中的mask标注图。目前我已经下载的DAUB和IRDST数据集中没有mask标注图的部分。您是下载了包含mask标注的IRDST数据集还是您自己生成制作了对应的mask图呢？此外，我尝试通过一些标注方法生成了IRDST和DAUB数据集的mask图，正巧我还没有对您原工作中的数据加载器部分进行调整，请问我能否直接使用您更新的IRDST数据集的加载器进行训练和测试？我是否还需要类似于您之前工作中的mat文件一样，进行一些额外的处理，比如添加部分噪声等。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

TinaLRJ commented 4 days ago

已更新了对背景不对齐数据的适配，欢迎重新下载