yolov7-face icon indicating copy to clipboard operation
yolov7-face copied to clipboard

Published 20 hours ago • verified publisher icon derronqi

utils/loss.py似乎不支持多分类问题

Open faccoco opened this issue 2 years ago • 3 comments
trafficstars

您好！我注意到在utils/loss.py的170行附近有这样一处表达：pkpt_score = ps[:, 8::3] 您似乎没有考虑多分类的情况？当分类数量大于1时，关键点的置信度应该并不是从8，而是从7+num_class处开始。

faccoco avatar Apr 05 '23 17:04 faccoco

您好！我注意到在utils/loss.py的170行附近有这样一处表达：pkpt_score = ps[:, 8::3] 您似乎没有考虑多分类的情况？当分类数量大于1时，关键点的置信度应该并不是从8，而是从7+num_class处开始。

请问为啥要从8改为7+num_calss呢

XhHello avatar Apr 13 '23 01:04 XhHello

您好！我注意到在utils/loss.py的170行附近有这样一处表达：pkpt_score = ps[:, 8::3] 您似乎没有考虑多分类的情况？当分类数量大于1时，关键点的置信度应该并不是从8，而是从7+num_class处开始。

那“pkpt_x = ps[:, 6::3] * 2. - 0.5” 和 “pkpt_y = ps[:, 7::3] * 2. - 0.5” 这两句是不是都要改，请问下您训练了多分类了吗？

CabbageWust avatar May 05 '23 08:05 CabbageWust

您好！我注意到在utils/loss.py的170行附近有这样一处表达：pkpt_score = ps[:, 8::3] 您似乎没有考虑多分类的情况？当分类数量大于1时，关键点的置信度应该并不是从8，而是从7+num_class处开始。

那“pkpt_x = ps[:, 6::3] * 2. - 0.5” 和 “pkpt_y = ps[:, 7::3] * 2. - 0.5” 这两句是不是都要改，请问下您训练了多分类了吗？

看起来是的。我还没能走完整个训练流程，看起来像是在生成混淆矩阵是出错了

faccoco avatar May 06 '23 13:05 faccoco