improving detection speed

thumbor · Nov 6, 2015 · c487097 · c487097
1 parent 143f0a2
commit c487097
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/remotecv/detectors/__init__.py b/remotecv/detectors/__init__.py
@@ -28,13 +28,19 @@ def load_cascade_file(self, module_path, cascade_file_path):
         cascade_file = join(abspath(dirname(module_path)), cascade_file_path)
         self.__class__.cascade = cv2.CascadeClassifier(cascade_file)
 
+    def get_min_size_for(self, size):
+        ratio = int(min(size[0], size[1]) / 15)
+        ratio = max(20, ratio)
+        return (ratio, ratio)
+
     def get_features(self, image):
         img = self.get_np_img(image)
 
         faces = self.__class__.cascade.detectMultiScale(
             img,
-            1.7,
-            1,
+            1.2,
+            4,
+            minSize=self.get_min_size_for(image.size)
         )
         faces_scaled = []
 

diff --git a/tests/fixtures/glasses.jpg b/tests/fixtures/glasses.jpg