There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks
Please refer to Assertion Error when training BodyPoseNet with my custom data.
BTW, since the source code is already open sourced, customer can modify and train a custom skeleton instead now.