License-Plate-Detector icon indicating copy to clipboard operation
License-Plate-Detector copied to clipboard

Published 20 hours ago • verified publisher icon zeusees

关于关键点数量

Open arthur0219 opened this issue 2 years ago • 6 comments

您好，感谢贵团队开源的工作。 1、请问是不是将deepcam-yolov5_face的代码改成了 4个 关键点检测的代码呢？我目前也是需要实现4个关键点检测，在尝试改yolov5-face的代码，恰好看到了您的工作 2、使用您的原始代码，在我自己的数据集上训练时，train的时候各个loss是正常下降的，但是val时targets map P R等所有指标为 0，您能提供一些解决思路吗

arthur0219 avatar May 14 '22 03:05 arthur0219

而且目前训了70个epoch的模型，加载之后进行detect，对于我自己的测试图像，能够正常检测出bbox和关键点，精度还蛮好，说明train时loss下降 模型参数得到了正确地优化。 但是不明白为什么val时 所有精度指标都为 0

arthur0219 avatar May 14 '22 03:05 arthur0219

1.我们就是基于deepcam-yolov5_face修改的四个点。 2.我记得val的时候，验证box的精度，没有验证四个点的回归精度。

Jinkham avatar May 14 '22 07:05 Jinkham

1.我们就是基于deepcam-yolov5_face修改的四个点。 2.我记得val的时候，验证box的精度，没有验证四个点的回归精度。

谢谢回复。 那么val的代码还需要改吗？我用您的代码直接训练的，在val的阶段 P R mAP这些都是 0

arthur0219 avatar May 14 '22 07:05 arthur0219

我也发现val阶段的指标为0的问题，你解决了吗？

tangsipeng avatar Jul 05 '22 02:07 tangsipeng

找到问题了，test 代码的 pred[:, 5] 改为 pred[:, 13] 就可以解决指标为0的问题，训练只用了loss，但是指标会影响最优模型的挑选，建议改过来比较好

tangsipeng avatar Jul 05 '22 08:07 tangsipeng

@tangsipeng 好的 谢谢建议

Jinkham avatar Jul 06 '22 11:07 Jinkham