diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_base_ap10k_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_base_ap10k_256x192.py index bd5daf5e..e3aa5d40 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_base_ap10k_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/ViTPose_base_ap10k_256x192.py @@ -127,30 +127,30 @@ test_pipeline = val_pipeline -data_root = 'data/apt36k' +data_root = 'data/ap10k' data = dict( - samples_per_gpu=32, + samples_per_gpu=64, workers_per_gpu=4, val_dataloader=dict(samples_per_gpu=32), test_dataloader=dict(samples_per_gpu=32), train=dict( type='AnimalAP10KDataset', - ann_file=f'{data_root}/annotations/train_annotations_1.json', - img_prefix=f'{data_root}/', + ann_file=f'{data_root}/annotations/ap10k-train-split1.json', + img_prefix=f'{data_root}/data/', data_cfg=data_cfg, pipeline=train_pipeline, dataset_info={{_base_.dataset_info}}), val=dict( type='AnimalAP10KDataset', - ann_file=f'{data_root}/annotations/val_annotations_1.json', - img_prefix=f'{data_root}/', + ann_file=f'{data_root}/annotations/ap10k-val-split1.json', + img_prefix=f'{data_root}/data/', data_cfg=data_cfg, pipeline=val_pipeline, dataset_info={{_base_.dataset_info}}), test=dict( type='AnimalAP10KDataset', - ann_file=f'{data_root}/annotations/val_annotations_1.json', - img_prefix=f'{data_root}/', + ann_file=f'{data_root}/annotations/ap10k-test-split1.json', + img_prefix=f'{data_root}/data/', data_cfg=data_cfg, pipeline=val_pipeline, dataset_info={{_base_.dataset_info}}), diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_base_apt36k_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_base_apt36k_256x192.py index e3aa5d40..bd5daf5e 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_base_apt36k_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/apt36k/ViTPose_base_apt36k_256x192.py @@ -127,30 +127,30 @@ test_pipeline = val_pipeline -data_root = 'data/ap10k' +data_root = 'data/apt36k' data = dict( - samples_per_gpu=64, + samples_per_gpu=32, workers_per_gpu=4, val_dataloader=dict(samples_per_gpu=32), test_dataloader=dict(samples_per_gpu=32), train=dict( type='AnimalAP10KDataset', - ann_file=f'{data_root}/annotations/ap10k-train-split1.json', - img_prefix=f'{data_root}/data/', + ann_file=f'{data_root}/annotations/train_annotations_1.json', + img_prefix=f'{data_root}/', data_cfg=data_cfg, pipeline=train_pipeline, dataset_info={{_base_.dataset_info}}), val=dict( type='AnimalAP10KDataset', - ann_file=f'{data_root}/annotations/ap10k-val-split1.json', - img_prefix=f'{data_root}/data/', + ann_file=f'{data_root}/annotations/val_annotations_1.json', + img_prefix=f'{data_root}/', data_cfg=data_cfg, pipeline=val_pipeline, dataset_info={{_base_.dataset_info}}), test=dict( type='AnimalAP10KDataset', - ann_file=f'{data_root}/annotations/ap10k-test-split1.json', - img_prefix=f'{data_root}/data/', + ann_file=f'{data_root}/annotations/val_annotations_1.json', + img_prefix=f'{data_root}/', data_cfg=data_cfg, pipeline=val_pipeline, dataset_info={{_base_.dataset_info}}),