yelusaleng / RRU-Net

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

训练时CASIA2.0里的mask的通道数为3 #29

Open laichou opened 2 years ago

laichou commented 2 years ago

您好,我在训练RRU时,将CASIA2.0里的1800+张splice的图片进行resize到384*256,也将.tif图片变成了.jpg图片。但是训练时,预测的结果通道是1,但是mask的通道为3,这导致计算loss的时候这两者的size不同,进而训练不了。您能告诉我这是什么原因吗? 非常感谢!

yelusaleng commented 2 years ago

说明你的训练图片的mask没有转换为单通道,mask = mask[:,:,0]就行了

laichou commented 2 years ago

非常感谢!我把所有的mask的位深度调整为了8,就可以正常训练了。

xiong76 commented 1 year ago

您好,我在训练RRU时,将CASIA2.0里的1800+张splice的图片进行resize到384*256,也将.tif图片变成了.jpg图片。但是训练时,预测的结果通道是1,但是mask的通道为3,这导致计算loss的时候这两者的size不同,进而训练不了。您能告诉我这是什么原因吗? 非常感谢!

请问你的tif文件是怎么转jpg进行训练的啊?我一直说找不到图片路径

laichou commented 1 year ago

自己写脚本转的

@.***

 

------------------ 原始邮件 ------------------ 发件人: "yelusaleng/RRU-Net" @.>; 发送时间: 2023年6月7日(星期三) 下午5:10 @.>; @.**@.>; 主题: Re: [yelusaleng/RRU-Net] 训练时CASIA2.0里的mask的通道数为3 (Issue #29)

您好,我在训练RRU时,将CASIA2.0里的1800+张splice的图片进行resize到384*256,也将.tif图片变成了.jpg图片。但是训练时,预测的结果通道是1,但是mask的通道为3,这导致计算loss的时候这两者的size不同,进而训练不了。您能告诉我这是什么原因吗? 非常感谢!

请问你的tif文件是怎么转jpg进行训练的啊?我一直说找不到图片路径

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

xiong76 commented 1 year ago

自己写脚本转的 酋 @.   ------------------ 原始邮件 ------------------ 发件人: "yelusaleng/RRU-Net" @.>; 发送时间: 2023年6月7日(星期三) 下午5:10 @.>; @*.**@*.>; 主题: Re: [yelusaleng/RRU-Net] 训练时CASIA2.0里的mask的通道数为3 (Issue #29) 您好,我在训练RRU时,将CASIA2.0里的1800+张splice的图片进行resize到384256,也将.tif图片变成了.jpg图片。但是训练时,预测的结果通道是1,但是mask的通道为3,这导致计算loss的时候这两者的size不同,进而训练不了。您能告诉我这是什么原因吗? 非常感谢! 请问你的tif文件是怎么转jpg进行训练的啊?我一直说找不到图片路径 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: **@.***>

我解决了这个问题之后,我发现CASIA数据集里面有些groundtruth和img对应不上,一直训练不起来,请问你怎么解决的啊

laichou commented 1 year ago

应该是CASIA里的有些图片命名错误导致对应不上,你可以网上查查。

@.***

 

------------------ 原始邮件 ------------------ 发件人: "yelusaleng/RRU-Net" @.>; 发送时间: 2023年6月8日(星期四) 下午2:18 @.>; @.**@.>; 主题: Re: [yelusaleng/RRU-Net] 训练时CASIA2.0里的mask的通道数为3 (Issue #29)

自己写脚本转的 酋 @.*   … ------------------ 原始邮件 ------------------ 发件人: "yelusaleng/RRU-Net" @.>; 发送时间: 2023年6月7日(星期三) 下午5:10 @.>; @.@.>; 主题: Re: [yelusaleng/RRU-Net] 训练时CASIA2.0里的mask的通道数为3 (Issue #29) 您好,我在训练RRU时,将CASIA2.0里的1800+张splice的图片进行resize到384256,也将.tif图片变成了.jpg图片。但是训练时,预测的结果通道是1,但是mask的通道为3,这导致计算loss的时候这两者的size不同,进而训练不了。您能告诉我这是什么原因吗? 非常感谢! 请问你的tif文件是怎么转jpg进行训练的啊?我一直说找不到图片路径 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.>

我解决了这个问题之后,我发现CASIA数据集里面有些groundtruth和img对应不上,一直训练不起来,请问你怎么解决的啊

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