Merge pull request #95 from Dobiasd/master

Fix UnboundLocalError in alignment_procedure, closes #94
serengil · Mar 20, 2024 · 059fcdf · 059fcdf
2 parents 29d593f + 6de5306
commit 059fcdf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/retinaface/commons/postprocess.py b/retinaface/commons/postprocess.py
@@ -87,6 +87,8 @@ def alignment_procedure(
 
         img = Image.fromarray(img)
         img = np.array(img.rotate(direction * angle))
+    else:
+        angle = 0.0  # Dummy value for undefined angle
 
     # -----------------------