dataset_maps = dict()
"""
each item in the dataset maps are a list of the following info
(
dataset_folder, 
annotation file name (video dataset) / path of annotation file (image dataset), 
split file name (video dataset) / path of image folder (image dataset) , 
modality
)
"""

dataset_maps['TAO'] = ['TAO',
                       'anno_person.json',
                       'splits_person.json',
                       'video']

dataset_maps['CRP'] = ['caltech_roadside_pedestrians',
                       'anno.json',
                       'splits.json',
                       'video']

dataset_maps['MOT17_DPM'] = ['MOT17',
                             'anno.json',
                             'splits_DPM.json',
                             'video']

dataset_maps['MOT17'] = ['MOT17',
                         'anno.json',
                         'splits.json',
                         'video']

dataset_maps['AOT'] = ['airbone_object_tracking',
                       'anno.json',
                       'splits.json',
                       'video']

dataset_maps['COCO17_train'] = ['mscoco',
                                'annotations/MSCOCO2017_train_person.json',
                                'images/train2017',   # all raw images would be in dataset_root/mscoco/images/train2017
                                'image']

dataset_maps['crowdhuman_train_fbox'] = ['CrowdHuman',
                                         'annotations/annotation_train_fbox.json',
                                         'Images',  # all raw images would be in dataset_root/CrowdHuman/Images
                                         'image']

dataset_maps['crowdhuman_train_vbox'] = ['CrowdHuman',
                                         'annotations/annotation_train_vbox.json',
                                         'Images',
                                         'image']

dataset_maps['crowdhuman_val_fbox'] = ['CrowdHuman',
                                       'annotations/annotation_val_fbox.json',
                                       'Images',
                                       'image']

dataset_maps['crowdhuman_val_vbox'] = ['CrowdHuman',
                                       'annotations/annotation_val_vbox.json',
                                       'Images',
                                       'image']

# Kristyna Janku: Added info for videos from CTC datasets Fluo-N2DH-SIM+ and BF-C2DL-MuSC
dataset_maps['Fluo-N2DH-SIM+_train_image_01'] = ['Fluo-N2DH-SIM+',
                                                 'annotation/Fluo-N2DH-SIM+_image_01.json',
                                                 'raw_data/train/01',
                                                 'image']

dataset_maps['Fluo-N2DH-SIM+_train_image_02'] = ['Fluo-N2DH-SIM+',
                                                 'annotation/Fluo-N2DH-SIM+_image_02.json',
                                                 'raw_data/train/02',
                                                 'image']

dataset_maps['Fluo-N2DH-SIM+_train_video_01_02'] = ['Fluo-N2DH-SIM+',
                                                    'N2DH_01_02.json',
                                                    'splits_N2DH_01_02.json',
                                                    'video']

dataset_maps['Fluo-N2DH-SIM+_test_video_01_02'] = ['Fluo-N2DH-SIM+',
                                                   'N2DH_01_02_test.json',
                                                   'splits_N2DH_01_02_test.json',
                                                   'video']


dataset_maps['BF-C2DL-MuSC_train_image_01'] = ['BF-C2DL-MuSC',
                                               'annotation/BF-C2DL-MuSC_image_01.json',
                                               'raw_data/train/01',
                                               'image']

dataset_maps['BF-C2DL-MuSC_train_video_01'] = ['BF-C2DL-MuSC',
                                               'MuSC_01.json',
                                               'splits_MuSC_01.json',
                                               'video']

dataset_maps['BF-C2DL-MuSC_train_video_02'] = ['BF-C2DL-MuSC',
                                               'MuSC_02.json',
                                               'splits_MuSC_02.json',
                                               'video']

dataset_maps['BF-C2DL-MuSC_train_video_01_02'] = ['BF-C2DL-MuSC',
                                                  'MuSC_01_02.json',
                                                  'splits_MuSC_01_02.json',
                                                  'video']

dataset_maps['BF-C2DL-MuSC_test_video_01_02'] = ['BF-C2DL-MuSC',
                                                 'MuSC_01_02_test.json',
                                                 'splits_MuSC_01_02_test.json',
                                                 'video']
