nsfw-resnet icon indicating copy to clipboard operation
nsfw-resnet copied to clipboard

关于训练

Open Usernamezhx opened this issue 4 years ago • 7 comments

你好 我的数据分4类,每类10w左右,我finetune 所有layer,batchsize 128,lr 0.05 我发现迭代2个epoch模型的val acc 就可以到0.98左右,没感觉到是过拟合,但是实际测试的时候效果不是很好,许多正常的图片也会被误识别为性感的图片而且置信度还很高,感觉很怪,难道是网络过拟合了?但是观察val上的表现又不像,不知道您有没有遇到过这种情况,感谢

Usernamezhx avatar Aug 24 '20 06:08 Usernamezhx

我出现过如下情况: 2个epoch后,val acc 90+。用测试集的样本去测试,准确率也会在85-90之间 你的被误识别的图片能发出来看看不?

yangbisheng2009 avatar Aug 26 '20 00:08 yangbisheng2009

你好,感谢你的回复,我分为4类:0:性感女性。1:赤裸男。2.普通女。3.其他(不包含人像)。我感觉网络学习到的东西可能有些偏差: 热图,训练集性感女: 屏幕快照 2020-08-26 下午1 51 36 热图,训练集普通女: 屏幕快照 2020-08-26 下午1 51 01 等会我再跑几个误识别的图传上来

Usernamezhx avatar Aug 26 '20 05:08 Usernamezhx

下边的是一些误召的,都预测为性感女,尤其最后一个的热图关注点不在胸部: GNqyPa6Hk0yE2NYgj0Jec3YkkDj1Nb cam 3omnx2piwjE5nyNp5LazTLoRnEJpE8 cam Q8Kwo3DHe69oZwDPdwJVTdVxOj0Jql ![cam](https://user-images.githubusercontent.com/14010625/91267569-410a4180-e7a6-11ea-8b00- cam 11d968b31678.jpg) XOx28J0TEkB48jEyWJgCJYPoxNYR9 oGewl33Hzm1Pd9omglRs39j5V2XG2 cam

Usernamezhx avatar Aug 26 '20 06:08 Usernamezhx

先确认下,你的训练集样本,都是原始图片吧(非热图)? 你列举的这几个性感误召回: 第一个:全身照,露腿,从特征上看,和你的普通女性训练集,相差比较远,更接近性感训练集的特征 第二个:这个可以算作是过拟合问题,也可以算作训练集样本问题,在训练的过程中,过分注重了裸露皮肤这一特征,你可以在普通女性训练集里面,适当加一下 第三个:和第一个类似,女性的身材,是一个重要特征,如果你的识别目标限于 裸露+身材or器官,可以适当的在普通女性样本集里加一些 穿衣服+身材的样本 第四个:识别是正确的吧

我现在的理解,模型的学习方向并没有偏,恰恰是按照你的训练集指导的特征来学的。训练集的对抗调整,要花一些时间的

yangbisheng2009 avatar Aug 26 '20 09:08 yangbisheng2009

你说的对,还是要对数据有深刻的理解,感谢

Usernamezhx avatar Oct 29 '20 12:10 Usernamezhx

火鸡 有试过使用检测的方法做吗?

Usernamezhx avatar Oct 29 '20 12:10 Usernamezhx

有的,百度信息流目前检测色情的方法,就是先做目标检测,然后再做色情、低俗分类 你可以试试,如果用目标检测,需要首先设置好待检测的类型: 1、人 2、物体,如:内衣、器械、器官等

yangbisheng2009 avatar Oct 30 '20 07:10 yangbisheng2009