HigherHRNet测试自己的数据集报错，如何解决？(语言-python)

问题遇到的现象和发生背景

HigherHRNet使用自己数据集（6个关键点），原作者使用coco（17个关键点），训练成功但测试失败。

用代码块功能插入代码，请勿粘贴截图

运行结果及报错内容

我的解答思路和尝试过的方法

将代码中所有17都改为6。
为保证数据无误，测试集使用的就是训练集。
标注数据工具使用coco-annotator。

我想要达到的结果

测试自己的数据集正常运行。

你这图有点糊啊。你训练的网络里面你得看下是不是还是17个点还是有参数修改成了6个点。
应该是设置的不对导致类别错误，并不是简单将代码中所有的17改成6就行，你看下模型读取的时候，模型的网络结构里面是17还是6？结构读取之后，你训练出来的网络权重赋值过去有无错误？