HRNet训练自己的数据集

Question

HRNet在进行姿态估计时，是一个top-down（自上而下的）算法，但我在官方源码中未找到需要加载目标检测权重的部分。想要对自己的数据集从头进行训练（从目标检测开始），但官方代码中并未有这部分，请问该怎么解决？

Answer 1

该回答引用GPTᴼᴾᴱᴺᴬᴵ
HRNet在进行姿态估计时确实是一个top-down算法，但它使用的目标检测器是单独训练的，并不需要在姿态估计时加载目标检测权重。因此，在官方代码中你也不会找到需要加载目标检测权重的部分。
-
如果你想要对自己的数据集从头进行训练（从目标检测开始），那么你需要先准备自己的目标检测数据集，并使用目标检测算法（如Faster R-CNN或YOLO）进行训练和验证。然后，你可以使用HRNet的代码对姿态估计进行训练和验证，只需要将你自己训练的目标检测器的权重文件路径作为输入参数传递给HRNet的训练代码即可。
-
具体来说，你需要在HRNet的代码中找到使用目标检测器进行姿态估计的部分，这部分代码应该在数据预处理和数据加载部分，通过调用目标检测器的代码来获取姿态估计器的输入数据。你可以根据你自己训练的目标检测器的代码和模型文件，修改HRNet的代码以适应你自己的数据集。