diff options
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 |