pyramids=2

2 files changed, 2 insertions, 2 deletions

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index b3bce9bc..61789fb9 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -66,7 +66,7 @@ def upload(dataset_name):
   detector = face_detector.DetectorDLIBHOG()
 
   # get detection as BBox object
-  bboxes = detector.detect(im_np, largest=True)
+  bboxes = detector.detect(im_np, largest=True, pyramids=2)
   if not bboxes or not len(bboxes):
     return jsonify({
       'error': 'bbox'
diff --git a/megapixels/app/server/tasks/demo.py b/megapixels/app/server/tasks/demo.py
index f7db9034..12d83383 100644
--- a/megapixels/app/server/tasks/demo.py
+++ b/megapixels/app/server/tasks/demo.py
@@ -81,7 +81,7 @@ def demo_task(self, uuid_name, fn):
   face_detector_instance = face_detector.DetectorDLIBCNN(gpu=opt_gpu)  # -1 for CPU
   step('Detecting face')
   st = time.time()
-  bboxes = face_detector_instance.detect(im_resized, largest=True)
+  bboxes = face_detector_instance.detect(im_resized, largest=True, pyramids=2)
   bbox = bboxes[0]
   dim = im_resized.shape[:2][::-1]
   bbox_dim = bbox.to_dim(dim)