Add a New Dataset - MichiganCOG/ViP GitHub Wiki

Add a New Dataset

To add a new dataset:

1. Convert annotation data to our JSON format
   • The JSON skeleton templates can be found here
   • Existing scripts for datasets can be found here
2. Create a dataset class in ViP/datasets/custom_dataset_name.py.
   • Inherit DetectionDataset or RecognitionDataset from ViP/abstract_dataset.py
   • Complete __init__ and __getitem__ functions
   • Example skeleton dataset can be found here