diff --git a/retinaface/commons/postprocess.py b/retinaface/commons/postprocess.py index c223508..8cbde32 100644 --- 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 # -----------------------