bugfix: Aligner size filter to int64

deepfakes · Oct 4, 2022 · 1f31af2 · 1f31af2
1 parent 6f48d7f
commit 1f31af2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/extract/align/_base.py b/plugins/extract/align/_base.py
@@ -607,7 +607,7 @@ def _scale_test(self,
         if self._min_scale <= 0.0 and self._max_scale <= 0.0:
             return None
 
-        roi = face.original_roi
+        roi = face.original_roi.astype("int64")
         size = ((roi[1][0] - roi[0][0]) ** 2 + (roi[1][1] - roi[0][1]) ** 2) ** 0.5
 
         if self._min_scale > 0.0 and size < minimum_dimension * self._min_scale: