Improving detection speed

thumbor · Nov 6, 2015 · 4d7722c · 4d7722c
1 parent 218d357
commit 4d7722c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/remotecv/detectors/__init__.py b/remotecv/detectors/__init__.py
@@ -8,7 +8,7 @@
 # http://www.opensource.org/licenses/mit-license
 # Copyright (c) 2011 globo.com timehome@corp.globo.com
 
-from os.path import join, dirname, abspath, isabs
+from os.path import join, dirname, abspath
 
 import cv2
 import numpy as np
@@ -25,10 +25,6 @@ def detect(self, context):
 class CascadeLoaderDetector(BaseDetector):
 
     def load_cascade_file(self, module_path, cascade_file_path):
-        # if not hasattr(self.__class__, 'cascade'):
-        #     if isabs(cascade_file_path):
-        #         cascade_file = cascade_file_path
-        #     else:
         cascade_file = join(abspath(dirname(module_path)), cascade_file_path)
         self.__class__.cascade = cv2.CascadeClassifier(cascade_file)
 
@@ -37,8 +33,8 @@ def get_features(self, image):
 
         faces = self.__class__.cascade.detectMultiScale(
             img,
-            1.2,
-            2,
+            1.7,
+            1,
         )
         faces_scaled = []
 

diff --git a/tests/test_image_processor.py b/tests/test_image_processor.py
@@ -15,7 +15,7 @@ def test_when_detector_unavailable(self):
 
     def test_when_image_is_huge(self):
         image_processor = ImageProcessor()
-        detect = image_processor.detect('face', read_fixture('huge_image.jpg'))
+        detect = image_processor.detect('all', read_fixture('huge_image.jpg'))
         expect(detect).Not.to_be_empty()
 
     def test_when_not_animated_gif(self):