Spaces:
Runtime error
Runtime error
File size: 1,375 Bytes
6b14fa5 65ed4c1 8fe1b94 a71f519 6b14fa5 65ed4c1 363a646 65ed4c1 363a646 65ed4c1 e91f073 363a646 e91f073 18f53a5 701d11a e91f073 f901f58 e91f073 f901f58 e91f073 103f82b f901f58 103f82b e91f073 f901f58 8fe1b94 65ed4c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import easyocr
import numpy as np
import cv2
import re
reader = easyocr.Reader(['en'], gpu=False)
def extract_weight_from_image(pil_img):
try:
img = np.array(pil_img)
# STEP 1: Resize and convert to grayscale
img = cv2.resize(img, None, fx=4, fy=4, interpolation=cv2.INTER_CUBIC)
gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
# STEP 2: Denoise + Threshold
blur = cv2.GaussianBlur(gray, (5, 5), 0)
_, thresh = cv2.threshold(blur, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
# Invert to get black text on white background
inverted = cv2.bitwise_not(thresh)
# STEP 3: OCR
results = reader.readtext(inverted)
# Debug print
print("OCR Results:", results)
# STEP 4: Extract weight values using regex
weight_candidates = []
for _, text, conf in results:
text = text.replace("kg", "").replace("KG", "").strip()
if re.match(r"^\d{2,4}(\.\d{1,2})?$", text):
weight_candidates.append((text, conf))
if not weight_candidates:
return "Not detected", 0.0
# STEP 5: Highest confidence
weight, confidence = sorted(weight_candidates, key=lambda x: -x[1])[0]
return weight, round(confidence * 100, 2)
except Exception as e:
return f"Error: {str(e)}", 0.0
|