ManTraNet-Pytorch icon indicating copy to clipboard operation
ManTraNet-Pytorch copied to clipboard

该如何训练其他的数据集

Open TsForever opened this issue 2 years ago • 9 comments

你好作者大大,目前我运行了你的代码,在NIST上训练,我应该怎么在CASIA或者其他数据集进行训练呢?我想将特征提取部分换成U-net改在哪里改呢?

TsForever avatar Oct 19 '23 08:10 TsForever

感谢你的关注,你需要自己定义一个专门给CASIA数据集的格式load的Dataset类,并且把路径导向该位置,可能需要涉及如下部分的修改:

  • 数据集路径: https://github.com/SunnyHaze/ManTraNet-Pytorch/blob/008b38329d9291e6fbcfb5695a7cade524dac575/Train.py#L51

  • Dataset class: https://github.com/SunnyHaze/ManTraNet-Pytorch/blob/008b38329d9291e6fbcfb5695a7cade524dac575/Train.py#L56

我们也提供了一个去掉了一部分“谬误”的CASIA数据集供参考,请在这里查看,如果喜欢可以给个star: Readme Card

然后你如果想修改特征提取的网络,则需要修改模型,这个需要修改model文件,大致在这个位置: https://github.com/SunnyHaze/ManTraNet-Pytorch/blob/008b38329d9291e6fbcfb5695a7cade524dac575/Mantra_Net.py#L28

希望能帮到你!

SunnyHaze avatar Oct 19 '23 09:10 SunnyHaze

之前也有一位朋友好像在CASIA数据集上跑过这个Mantra-Net的复现,你也可以看看有没有机会咨询一下他:

https://github.com/SunnyHaze/ManTraNet-Pytorch/issues/3#issuecomment-1109320281

SunnyHaze avatar Oct 19 '23 09:10 SunnyHaze

之前也有一位朋友好像在CASIA数据集上跑过这个Mantra-Net的复现,你也可以看看有没有机会咨询一下他:

#3 (comment)

好的,谢谢你,但是我看他主页没有联系方式(哭哭)

TsForever avatar Oct 19 '23 10:10 TsForever

好滴👌,不过我感觉那位朋友可能不太会及时回复,不知道您想用CASIA训练是出于什么动机?如果是初入科研的练手项目的话,我个人建议还是比较值得自己啃一下实现的,因为换一个新数据集确实是整个篡改检测任务最基础也最经常遇到的操作,对齐好数据路径以及转换的格式即可。对于Pytorch的DatasetDataloader的接口和实现也有很多的CSDN和知乎教程可以参考。也欢迎随时交流。

SunnyHaze avatar Oct 19 '23 10:10 SunnyHaze

你好作者大大,目前我运行了你的代码,在NIST上训练,我应该怎么在CASIA或者其他数据集进行训练呢?我想将特征提取部分换成U-net改在哪里改呢?

这个代码里面读取的是train和test,但是nist16数据集不是三个类型的吗,能分享一下你训练的这个nist16数据集吗?非常感谢

xiong76 avatar Nov 23 '23 07:11 xiong76

你好作者大大,目前我运行了你的代码,在NIST上训练,我应该怎么在CASIA或者其他数据集进行训练呢?我想将特征提取部分换成U-net改在哪里改呢?

这个代码里面读取的是train和test,但是nist16数据集不是三个类型的吗,能分享一下你训练的这个nist16数据集吗?非常感谢

或许可以加个联系方式交流交流一下吗

xiong76 avatar Nov 23 '23 07:11 xiong76

NIST 的三个类型应该是Splicing? manipulated还有个xxxx(我记不住了),但其实manipulated是另外两个的“和”。所以实质上只有manipulated用于训练。

这个仓库里面的train和test是我从manipulated通过8:2随机分配得到的。

可以联系,为了保密,你可以邮件给我你的qq或者微信号,我的工作邮箱在我的github profile.

SunnyHaze avatar Nov 23 '23 07:11 SunnyHaze

非常感谢,但是我怎么没有看见工作邮箱📪😂

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: Ma Xiaochen (马晓晨) @.> 发送时间: 2023年11月23日 15:11 收件人: SunnyHaze/ManTraNet-Pytorch @.> 抄送: null_xx @.>, Comment @.> 主题: Re: [SunnyHaze/ManTraNet-Pytorch] 该如何训练其他的数据集 (Issue #5)

xiong76 avatar Nov 23 '23 07:11 xiong76