how to output the result in coco json file and retrain a new model about simple-hrnet HOT 2 CLOSED

Is there a possible way just to use the keypoint as the G.T label?

from simple-hrnet.

Hi, I'm happy you found the project useful!

There was some discussion in the past on how to train on custom datasets (see issue #56 ).

I'm not aware of any easy way to convert the network outputs in the COCO dataset format, but you can save them in any format and create a custom training dataset that loads them, keeping in mind that

Basically, if you want to use the train_coco.py script, you need a getitem method that returns a tuple containing
image, target, target_weight, joints_data and a evaluate_accuracy method that returns accs, avg_acc, cnt, joints_preds, joints_target.

It should be relatively simple and much easier than trying to format your dataset in the COCO format imho.

Let me know if that helps 🙂

from simple-hrnet.