avoid indentation

3 weeks ago · c3163caefd
parent 055463d23a
commit c3163caefd
1 changed files with 213 additions and 214 deletions
--- a/src/eynollah/eynollah.py
+++ b/src/eynollah/eynollah.py
@ -1140,9 +1140,8 @@ class Eynollah:
            seg_color = np.repeat(seg[:, :, np.newaxis], 3, axis=2)
            prediction_true = resize_image(seg_color, img_h_page, img_w_page)
            prediction_true = prediction_true.astype(np.uint8)
+            return prediction_true

-
-        else:
        if img.shape[0] < img_height_model:
            img = resize_image(img, img_height_model, img.shape[1])