ocr updated

Layoutlmv3_inference/ocr.py

@@ -63,11 +63,7 @@ def enhance_txt(img, intensity_increase=20, bilateral_filter_diameter=9, bilater
     # Convert image to grayscale
     grayscale_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 
-    # Find contours
-    contours, _ = cv2.findContours(
-        grayscale_img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-
-    # # Apply Gaussian blur
+    # Apply Gaussian blur
     blurred = cv2.GaussianBlur(grayscale_img, (1, 1), 0)
 
     edged = 255 - cv2.Canny(blurred, 100, 150, apertureSize=7)
@@ -76,8 +72,7 @@ def enhance_txt(img, intensity_increase=20, bilateral_filter_diameter=9, bilater
     img = np.clip(img + intensity_increase, 0, 255).astype(np.uint8)
 
     # Apply bilateral filter to reduce noise
-    img = cv2.bilateralFilter(img, bilateral_filter_diameter,
-                              bilateral_filter_sigma_color, bilateral_filter_sigma_space)
+    img = cv2.bilateralFilter(img, bilateral_filter_diameter, bilateral_filter_sigma_color, bilateral_filter_sigma_space)
 
     _, binary = cv2.threshold(blurred, threshold, 255, cv2.THRESH_BINARY)
     return binary