diff --git a/im_generator.py b/im_generator.py index 4f4c0bf..ebe6491 100644 --- a/im_generator.py +++ b/im_generator.py @@ -51,7 +51,7 @@ def remove_background(img): img[thresh == 255] = 0 - kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (13, 13)) + kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (17, 17)) erosion = cv2.erode(img, kernel, iterations=1) erosion_gray = cv2.cvtColor(erosion, cv2.COLOR_BGR2GRAY)