HRNet 在mpii数据集上能识别出16个关键点,用预训练好的HRNet 网络去测试自定义关键点,但只想得到其中14个关键点,怎么把多余的两个关键点去掉
时间: 2024-04-21 12:27:04 浏览: 108
基于ONNXRuntime部署PP-YOLOE行人检测+HRNet人体骨骼关键点检测源码+模型.zip
如果你只需要其中14个关键点,可以在训练模型时将这两个关键点的标注数据去掉。如果你已经训练好了模型,可以通过修改模型的输出层来实现只输出14个关键点的坐标。具体来说,你需要修改模型输出层的大小,使其只输出14个关键点的坐标,并且在训练数据和测试数据中也只保留14个关键点的标注数据。如果你使用的是现成的预训练模型,可以尝试修改模型的配置文件或者代码,以适应你的需要。
阅读全文