barc_gradio / src /configs /dataset_path_configs.py
Nadine Rueegg
initial commit for barc
7629b39
raw
history blame
1.12 kB
import numpy as np
import os
import sys
abs_barc_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..',))
# stanext dataset
# (1) path to stanext dataset
STAN_V12_ROOT_DIR = abs_barc_dir + '/datasets/StanfordExtra_V12/'
IMG_V12_DIR = os.path.join(STAN_V12_ROOT_DIR, 'StanExtV12_Images')
JSON_V12_DIR = os.path.join(STAN_V12_ROOT_DIR, 'labels', "StanfordExtra_v12.json")
STAN_V12_TRAIN_LIST_DIR = os.path.join(STAN_V12_ROOT_DIR, 'labels', 'train_stanford_StanfordExtra_v12.npy')
STAN_V12_VAL_LIST_DIR = os.path.join(STAN_V12_ROOT_DIR, 'labels', 'val_stanford_StanfordExtra_v12.npy')
STAN_V12_TEST_LIST_DIR = os.path.join(STAN_V12_ROOT_DIR, 'labels', 'test_stanford_StanfordExtra_v12.npy')
# (2) path to related data such as breed indices and prepared predictions for withers, throat and eye keypoints
STANEXT_RELATED_DATA_ROOT_DIR = os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'stanext_related_data')
# image crop dataset (for demo, visualization)
TEST_IMAGE_CROP_ROOT_DIR = os.path.join(os.path.dirname(__file__), '..', '..', 'datasets', 'test_image_crops')