diff options
| author | adamhrv <adam@ahprojects.com> | 2019-02-14 14:45:18 +0100 |
|---|---|---|
| committer | adamhrv <adam@ahprojects.com> | 2019-02-14 14:45:18 +0100 |
| commit | 3a3a89f2c58eceee07b2cfcfb1700a61b34619e5 (patch) | |
| tree | 436347b8466422a1019209f9f04937ea1ce0e4eb /megapixels/app/settings | |
| parent | 41247c08ea359d0a72a247992d2019ae2120536c (diff) | |
updates
Diffstat (limited to 'megapixels/app/settings')
| -rw-r--r-- | megapixels/app/settings/app_cfg.py | 7 | ||||
| -rw-r--r-- | megapixels/app/settings/types.py | 5 |
2 files changed, 9 insertions, 3 deletions
diff --git a/megapixels/app/settings/app_cfg.py b/megapixels/app/settings/app_cfg.py index 2b10f9f0..0b1fb69d 100644 --- a/megapixels/app/settings/app_cfg.py +++ b/megapixels/app/settings/app_cfg.py @@ -6,6 +6,7 @@ from dotenv import load_dotenv from app.settings import types from app.utils import click_utils +from pathlib import Path import codecs codecs.register(lambda name: codecs.lookup('utf8') if name == 'utf8mb4' else None) @@ -26,6 +27,10 @@ FaceLandmark2D_5Var = click_utils.ParamVar(types.FaceLandmark2D_5) FaceLandmark2D_68Var = click_utils.ParamVar(types.FaceLandmark2D_68) FaceLandmark3D_68Var = click_utils.ParamVar(types.FaceLandmark3D_68) +# base path +DIR_SELF = os.path.dirname(os.path.realpath(__file__)) +DIR_ROOT = Path(DIR_SELF).parent.parent.parent + # # data_store DATA_STORE = '/data_store_hdd/' DATA_STORE_NAS = '/data_store_nas/' @@ -64,7 +69,7 @@ DIR_TEST_IMAGES = join(DIR_APP, 'test', 'images') # ----------------------------------------------------------------------------- # .env config for keys # ----------------------------------------------------------------------------- - +FP_KNOWLEDGE_GRAPH_ENV = join(DIR_ROOT, 'env/google_knowledge_graph_api.env') # DIR_DOTENV = join(DIR_APP, '.env') load_dotenv() # dotenv_path=DIR_DOTENV) diff --git a/megapixels/app/settings/types.py b/megapixels/app/settings/types.py index 933d1932..3d7e96c0 100644 --- a/megapixels/app/settings/types.py +++ b/megapixels/app/settings/types.py @@ -47,8 +47,9 @@ class Metadata(Enum): FACE_ATTRIBUTES, IMAGE_COUNT = range(10) class Dataset(Enum): - LFW, VGG_FACE2, MSCELEB, UCCS, UMD_FACES, SCUT_FBP, UCF_SELFIE, UTK, \ - CASIA_WEBFACE, AFW, PUBFIG83, HELEN, PIPA, MEGAFACE, BRAINWASH, IMDB_WIKI = range(16) + LFW, VGG_FACE, VGG_FACE2, MSCELEB, UCCS, UMD_FACES, SCUT_FBP, UCF_SELFIE, UTK, \ + CASIA_WEBFACE, AFW, PUBFIG83, HELEN, PIPA, MEGAFACE, BRAINWASH, IMDB_WIKI, \ + LARGE_AGE_GAP = range(18) # --------------------------------------------------------------------- |