yelusaleng / RRU-Net

Official repository for "RRU-Net: The Ringed Residual U-Net for Image Splicing Forgery Detection" (CVPRW 2019)
106 stars 17 forks source link

关于训练集和测试集的选择? #23

Closed 170744039 closed 2 years ago

170744039 commented 2 years ago

尊敬的作者您好,我知道回复问题也可能是一件比较烦人的事情。但是我不得已需要就这个问题请教一下您。在CISIA2.0中有大概一千八百多张拼接篡改,但您并没有全部采用,经过我的实验,选取适合的训练集和测试集对最终的结果有相当大影响。因为我迫切需要复现您的结果,所以我想请教一下您的训练集和测试集是否可以公布,以文本列表的形式也可以,如果有些麻烦,是否可以告诉我您是怎样选取715和这个100张图片的。

170744039 commented 2 years ago

补充一下,通过使用您的best_model,在我随机选取的100张测试集上可以达到 acc :0.94001359 recall :0.98047918 f1 :0.95415276 这样非常好的结果,而我训练的结果却只有 f1 : 0.54 对于您是怎样训练的,我更加好奇了,我是严格按照您的源码进行的训练,这里面应该没有用论文中提到的学习率动态下降。而关于这个最佳模型best_model,我还想多请教一下是否有在整个cisia上进行微调?

yelusaleng commented 2 years ago

我分开回答你的两个问题,首先是第一个问题。 如你所说,经挑选的训练集和测试集对模型确实有很大的影响。实际上rru-net里所用到的715和100张图片是我直接沿用了我的前一个工作"Image splicing forgery detection combining coarse to refined convolutional neural network and adaptive clustering"里的数据集,因为时间久远,现在我已经找不到这些数据了。 不知道你为何要复现rru-net里的结果呢?以目前的研究进度来看,我当初的实验内容是放现在已经算是实验不足了。

对于我所提供的best_model,我在README.MD文件里已解释过,这是我新工作的所提出方法"Image splicing forgery detection by combining synthetic adversarial networks and hybrid dense U‐net based on multiple spaces"所生成的数据集来训练rru-net得到的模型。与学习率无关,与数据集有关。不过,生成的数据集也是由casia 2.0和forensics这两个数据集生成的,所以一定程度上也算是"微调"了吧。,具体的内容可以参考上面的文献。

170744039 commented 2 years ago

非常感谢您的及时回复,对于您提及的这两篇论文我还没有阅读过,可能还是有些疏忽。倒不是非要复现,训练集与测试集的选择无非是想要一个更好的结果,我自己的工作肯定是想要首先和您站在同一个起跑线上才行。如果有完整的,无需择优的数据集肯定是最为简单实用的,这些我可能还没有发言权,我先研读一下您的新工作所采用的方法,看看是否可以重现这个best_model,可能还会有叨扰,我先说声抱歉以及再次感谢您的回复。

yelusaleng commented 2 years ago

没事,有问题随时联系