From 8afe7455067996bf8bca087418425c2bb7d6d492 Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期三, 23 三月 2022 12:16:23 +0800
Subject: [PATCH] add height check
---
extract_voltage_task.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/extract_voltage_task.py b/extract_voltage_task.py
index cf30f1a..be8e3e9 100644
--- a/extract_voltage_task.py
+++ b/extract_voltage_task.py
@@ -56,7 +56,7 @@
max_val = cv2.minMaxLoc(res)[1]
source.append(max_val)
# print(max(source))
- if max(source) > 0.5:
+ if max(source) > 0.3:
return digits[source.index(max(source))][1]
else:
return "@"
@@ -69,14 +69,14 @@
cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU)[1]
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (6, 6))
counter = cv2.morphologyEx(thresh.copy(), cv2.MORPH_OPEN, kernel)
- #image_h = image.shape[0]
contours, hierarchy = cv2.findContours(counter, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
return_numbers = []
contours_count = 0
+ image_h = image.shape[0]
for c in contours:
x, y, w, h = cv2.boundingRect(c)
#if image_h*0.98 > h > image_h * 0.3:
- if hierarchy[0][contours_count][3] ==0 :
+ if hierarchy[0][contours_count][3] ==0 and h > image_h * 0.3:
#print((x, y, w, h))
#cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)
one_number = get_most_simmilar_digit(cv2.threshold(cv2.resize(
--
Gitblit v1.9.1