diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-02 00:24:10 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-02 00:24:10 +0100 |
| commit | 5fea322e763540e0861a735b0231d393aae47f04 (patch) | |
| tree | fde574682be98f55269b2160af6c8617e56a4c63 /cli/app/settings | |
| parent | 2ae7ffce3e374d5d28b8ddfefb347d04ca0f6e48 (diff) | |
fetch script
Diffstat (limited to 'cli/app/settings')
| -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') |