Pytorch_Retinaface icon indicating copy to clipboard operation
Pytorch_Retinaface copied to clipboard

Published 20 hours ago • verified publisher icon biubug6

你好，关于训练自己的数据集

Open ybcc2015 opened this issue 4 years ago • 0 comments
trafficstars

你好，我自己的数据集只有3个关键点，所以在WiderFaceDetection类中将annotations改成了11维（4个box坐标 + 6个关键点坐标 + 1个label），但在训练时总是报一些维度转换的错误，看了下代码，里面有很多地方都写死了5个关键点，请问有什么好方法只训练3个关键点吗？期待回复

ybcc2015 avatar Dec 31 '20 00:12 ybcc2015