merge

1 files changed, 14 insertions, 1 deletions

diff --git a/megapixels/app/processors/face_detector.py b/megapixels/app/processors/face_detector.py
index 02d068dc..747e057b 100644
--- a/megapixels/app/processors/face_detector.py
+++ b/megapixels/app/processors/face_detector.py
@@ -11,13 +11,26 @@ from app.utils import im_utils, logger_utils
 from app.models.bbox import BBox
 from app.settings import app_cfg as cfg
 
+
+class DetectorHaar:
+
+  im_size = (400, 400)
+  cascade_name = types.HaarCascade.FRONTAL
+
+  def __init__(self, cascade=types.HaarCascade.FRONTAL):
+    self.log = logger_utils.Logger.getLogger()
+
+  def detect(self, im, scale_factor=1.05, overlaps=5):
+    return
+    
+
 class DetectorDLIBCNN:
 
   dnn_size = (300, 300)
   pyramids = 0
   conf_thresh = 0.85
 
-  def __init__(self, opt_gpu):
+  def __init__(self, opt_gpu=0):
     self.log = logger_utils.Logger.getLogger()
     cuda_visible_devices = os.getenv('CUDA_VISIBLE_DEVICES', '')
     os.environ['CUDA_VISIBLE_DEVICES'] = str(opt_gpu)