From 5fea322e763540e0861a735b0231d393aae47f04 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 2 Dec 2019 00:24:10 +0100 Subject: fetch script --- cli/app/settings/app_cfg.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'cli/app/settings') 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') -- cgit v1.2.3-70-g09d2