Loading ballontranslator/dl/inpaint/lama.py +32 −28 Original line number Diff line number Diff line Loading @@ -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) Loading Loading
ballontranslator/dl/inpaint/lama.py +32 −28 Original line number Diff line number Diff line Loading @@ -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) Loading