Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于关键点数量 #60

Open
arthur0219 opened this issue May 14, 2022 · 6 comments
Open

关于关键点数量 #60

arthur0219 opened this issue May 14, 2022 · 6 comments

Comments

@arthur0219
Copy link

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

@arthur0219
Copy link
Author

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

@Jinkham
Copy link
Member

Jinkham commented May 14, 2022

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

@arthur0219
Copy link
Author

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

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

@tangsipeng
Copy link

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

@tangsipeng
Copy link

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

@Jinkham
Copy link
Member

Jinkham commented Jul 6, 2022

@tangsipeng 好的 谢谢建议

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants