From 592d0f7477c9a89fe81296f152a42df083e33ea6 Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期三, 23 三月 2022 15:06:51 +0800
Subject: [PATCH] add check return_numbers
---
extract_voltage_task.py | 2 ++
main.py | 2 ++
2 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/extract_voltage_task.py b/extract_voltage_task.py
index be8e3e9..beb438a 100644
--- a/extract_voltage_task.py
+++ b/extract_voltage_task.py
@@ -85,6 +85,8 @@
return_numbers.append([x, one_number])
else:
return return_numbers, False
+ if len(return_numbers)==0:
+ return 0,False
digit = 0
return_numbers = np.array(sorted(return_numbers))
return_numbers = return_numbers[:, -1]
diff --git a/main.py b/main.py
index 777886e..2ea84e7 100644
--- a/main.py
+++ b/main.py
@@ -143,6 +143,8 @@
cv2.imwrite("image_frame_"+time_now+".jpg", cv2.putText(frame, str(return_digital), (app.config['IMAGE_CLIP_ROWS_START'], app.config['IMAGE_CLIP_COLS_START']), cv2.FONT_HERSHEY_COMPLEX, 1.5, (0, 255, 255), 2))
cap.release()
break
+ else:
+ cv2.imwrite("./fails/image_frame_"+time_now+".jpg", frame)
#if need_send _image :
# send_image
--
Gitblit v1.9.1