xialuxi / yolov5_face_landmark

基于yolov5的人脸检测,带关键点检测
GNU General Public License v3.0
238 stars 53 forks source link

关于训练的一些问题? #17

Open Hezhexi2002 opened 3 years ago

Hezhexi2002 commented 3 years ago

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