Use LRU cache for shape transformation

Only cache transform if shapes are actually the same
ORB-HD · Jul 4, 2023 · 7082b35 · 7082b35
1 parent 814e6a2
commit 7082b35
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deface/centerface.py b/deface/centerface.py
@@ -1,6 +1,7 @@
-import datetime
 import os
 
+from functools import lru_cache
+
 import numpy as np
 import cv2
 
@@ -121,6 +122,7 @@ def __call__(self, img, threshold=0.5):
         return dets, lms
 
     @staticmethod
+    @lru_cache(maxsize=128)
     def shape_transform(in_shape, orig_shape):
         h_orig, w_orig = orig_shape
         w_new, h_new = in_shape