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

能否最后只检测四个角点的坐标?

Open Hezhexi2002 opened this issue 2 years ago • 7 comments

你好,大佬,我是一名大二学生,目前在我们学校robomaster战队里负责用神经网络识别装甲板,但是用yolo系列的话最后的bbox并不能很好的拟合装甲板的轮廓,导致我们后期在使用pnp进行姿态解算时会有很大误差,所以我想使用关键点检测的方法得到的bbox效果要好很多,就用装甲板的四个角点,就是我看您的这个项目应该能够实现,我想问一下就是需要改动哪些地方呢,loss和head就行了吗?

Hezhexi2002 avatar Oct 22 '21 07:10 Hezhexi2002

如果bbox效果都不是很好的话,用关键点回归应该也不会取得更好的效果,你可以试试,参考https://github.com/deepcam-cn/yolov5-face这个链接,检测+回归关键点,看看是否有提高。

Jinkham avatar Oct 23 '21 08:10 Jinkham

如果bbox效果都不是很好的话,用关键点回归应该也不会取得更好的效果,你可以试试,参考https://github.com/deepcam-cn/yolov5-face这个链接,检测+回归关键点,看看是否有提高。

谢谢大佬,这个仓库我其实也已经star了,但是用的话,还是需要改动loss以及head这些吗?

Hezhexi2002 avatar Oct 24 '21 02:10 Hezhexi2002

不需要

Jinkham avatar Oct 24 '21 02:10 Jinkham

不需要

大佬,就是我现在的数据集单张图片的标签格式是第1个为类别,后面8个分别是装甲板四个角点的归一化后的x,y坐标,意思是能够直接用那个仓库里的代码训练然后推理了吗?

Hezhexi2002 avatar Oct 24 '21 11:10 Hezhexi2002

要改代码,原始代码5个关键点,你这只有四个,要修改一下,不复杂。

Jinkham avatar Oct 25 '21 08:10 Jinkham

要改代码,原始代码5个关键点,你这只有四个,要修改一下,不复杂。

大佬,是主要就修改loss和head吗?

Hezhexi2002 avatar Oct 25 '21 12:10 Hezhexi2002

要改代码,原始代码5个关键点,你这只有四个,要修改一下,不复杂。

大佬,是主要就修改loss和head吗?

主要是我后期是想用c++部署,用trt或者openvint加速,现在我用的tensorrtx这个仓库在win下使用trt加速yolov5,就是不知道如果用这个模型也能用c+++部署吗?

Hezhexi2002 avatar Oct 25 '21 12:10 Hezhexi2002