diff options
| author | jules@lens <julescarbon@gmail.com> | 2019-04-18 16:55:14 +0200 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2019-04-18 16:55:14 +0200 |
| commit | 2e4daed06264f3dc3bbabd8fa4fc0d8ceed4c5af (patch) | |
| tree | 1a17bb4459776ac91f7006a2a407ca12edd3471e /megapixels/app/settings | |
| parent | 3d32e5b4ddbfbfe5d4abeda57ff200adf1532f4c (diff) | |
| parent | f8012f88641b0bb378ba79393f277c8918ebe452 (diff) | |
Merge branch 'master' of asdf.us:megapixels_dev
Diffstat (limited to 'megapixels/app/settings')
| -rw-r--r-- | megapixels/app/settings/app_cfg.py | 5 | ||||
| -rw-r--r-- | megapixels/app/settings/types.py | 3 |
2 files changed, 8 insertions, 0 deletions
diff --git a/megapixels/app/settings/app_cfg.py b/megapixels/app/settings/app_cfg.py index 1eed1a41..98d36b5f 100644 --- a/megapixels/app/settings/app_cfg.py +++ b/megapixels/app/settings/app_cfg.py @@ -19,6 +19,7 @@ LogLevelVar = click_utils.ParamVar(types.LogLevel) MetadataVar = click_utils.ParamVar(types.Metadata) DatasetVar = click_utils.ParamVar(types.Dataset) DataStoreVar = click_utils.ParamVar(types.DataStore) + # Face analysis HaarCascadeVar = click_utils.ParamVar(types.HaarCascade) FaceDetectNetVar = click_utils.ParamVar(types.FaceDetectNet) @@ -27,6 +28,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) +# Person/Body detector +BodyDetectNetVar = click_utils.ParamVar(types.BodyDetectNet) + + # base path DIR_SELF = os.path.dirname(os.path.realpath(__file__)) DIR_ROOT = Path(DIR_SELF).parent.parent.parent diff --git a/megapixels/app/settings/types.py b/megapixels/app/settings/types.py index 3d7e96c0..2609ece7 100644 --- a/megapixels/app/settings/types.py +++ b/megapixels/app/settings/types.py @@ -59,6 +59,9 @@ class FaceDetectNet(Enum): """Scene text detector networks""" HAAR, DLIB_CNN, DLIB_HOG, CVDNN, MTCNN_TF, MTCNN_PT, MTCNN_CAFFE = range(7) +class BodyDetectNet(Enum): + CVDNN = range(1) + class FaceExtractor(Enum): """Type of face recognition feature extractor""" # TODO deprecate DLIB resnet and use only CVDNN Caffe models |