diff options
Diffstat (limited to 'megapixels/app/server')
| -rw-r--r-- | megapixels/app/server/api.py | 2 | ||||
| -rw-r--r-- | megapixels/app/server/tasks/demo.py | 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) |