Commit 2b677ae6 authored by dmMaze's avatar dmMaze
Browse files

Fix lama_mpe infinity loop

parent 458ca100
Loading
Loading
Loading
Loading
+32 −28
Original line number Diff line number Diff line
@@ -367,6 +367,10 @@ class LamaFourier:
        pos = np.zeros((h, w), dtype=np.int32)
        direct = np.zeros((h, w, 4), dtype=np.int32)
        i = 0

        if mask3.max() > 0:
            # otherwise it will cause infinity loop
        
            while np.sum(1 - mask3) > 0:
                i += 1
                mask3_ = cv2.filter2D(mask3, -1, ones_filter)