yolov5-face

yolov5-face copied to clipboard

调整下hyp里面的权重。

Originally posted by @derronqi in https://github.com/deepcam-cn/yolov5-face/issues/99#issuecomment-982254928

我的做法你可以参考一下，效果还不错 1.增加test函数的输出，赠加人脸关键点的衡量指标NME 2.调整loss里landmark loss 权重 3.调整fitness 保存模型时NME指标的的权重

1 增加test的输出？这个文件不是再验证时使用的吗 增加nme的意义是啥呢。保存模型时以nme越小的原则？ 2 调整loss权重是一个方案，我还没有尝试，现在时增加较多的大尺度的人脸关键点数据，但会拉下检测的性能。 3 在1的基础上完成是吧。感谢大佬指点迷津。

能分享一份代码吗

代码不方便分享，你可以尝试自己改下，还很简单的。 1.因为test函数时在做验证和测试的作用，可以观察mAP下降的情况下，NME的变动，同时可以保存模型时以nme越小的原则 2 调整loss权重方案，可以修改data/hyp.scratch.yaml文件13～15行 3.是这样的

好的 感谢大佬指导方向 对这份代码不太熟悉,还是有些问题 1 怎么去掉训练时的验证过程,我只根据landmark 的损失来保存模型.或者epoch结束直接保存模型 2 请问你那边最佳的权重大概是什么样的调整规则,跟box一致吗? 3 test文件哪个地方在输出landmark呢. 4 多尺度训练 有尝试吗,对landmark有提点吗?