search

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
http://mvig.org/research/alphapose.html
Other
7.86k stars 1.96k forks source link

KeyError: 'CustomDataset is not in the dataset registry' #1116

Open LEIJUNMIUI opened 1 year ago

LEIJUNMIUI commented 1 year ago

how to solve this problem , I want train my datasets, I labeled it with coco format ,it include only 3 keypoints? thanks

nistring commented 1 year ago

In order to train custom dataset, you'll need to make your custom dataset file in alphapose/datasets. Assuming that your dataset follows coco format, make your_custom_dataset.py copying from mscoco.py then modify parameters such as EVAL_JOINTS num_joints 'joint_pairs. Lastly, enlist your custom dataset file in __init__.py. This should make the custom dataset recognizable if you properly assign the name of dataset in config file.