Skip to content

Commit 6de5306

Browse files
committed
Fix UnboundLocalError in alignment_procedure, closes #94
1 parent 29d593f commit 6de5306

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

retinaface/commons/postprocess.py

+2
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,8 @@ def alignment_procedure(
8787

8888
img = Image.fromarray(img)
8989
img = np.array(img.rotate(direction * angle))
90+
else:
91+
angle = 0.0 # Dummy value for undefined angle
9092

9193
# -----------------------
9294

0 commit comments

Comments
 (0)