diff options
Diffstat (limited to 'cli/app/settings/app_cfg.py')
| -rw-r--r-- | cli/app/settings/app_cfg.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/cli/app/settings/app_cfg.py b/cli/app/settings/app_cfg.py index fe65458..6f99fd0 100644 --- a/cli/app/settings/app_cfg.py +++ b/cli/app/settings/app_cfg.py @@ -26,7 +26,8 @@ CLICK_GROUPS = { SELF_CWD = os.path.dirname(os.path.realpath(__file__)) # Script CWD DIR_APP = str(Path(SELF_CWD).parent.parent.parent) - +DIR_IMAGENET = join(DIR_APP, 'data_store/imagenet') +DIR_OUTPUTS = join(DIR_APP, 'data_store/outputs') FP_MODELZOO = join(DIR_APP, 'modelzoo/modelzoo.yaml') # ----------------------------------------------------------------------------- @@ -35,3 +36,11 @@ FP_MODELZOO = join(DIR_APP, 'modelzoo/modelzoo.yaml') with open(FP_MODELZOO, 'r') as fp: MODELZOO_CFG = yaml.load(fp, Loader=yaml.Loader) + +# ----------------------------------------------------------------------------- +# Imagenet +# ----------------------------------------------------------------------------- + +IMAGENET_IMAGES_PER_CLASS = 200 +FP_IMAGENET_WORDS = join(DIR_IMAGENET, 'words.txt') +FP_IMAGENET_CLASSES = join(DIR_IMAGENET, 'classes_in_imagenet.csv') |