From e1b52edc09daf69bd5253a965d004014ae62752e Mon Sep 17 00:00:00 2001
From: HJW <1576345902@qq.com>
Date: Sat, 26 Oct 2024 13:50:05 +0800
Subject: [PATCH] =?UTF-8?q?UPDATE=20Vision=20=E6=9B=B4=E6=96=B0=E8=A1=8C?=
 =?UTF-8?q?=E4=BA=BA=E6=A3=80=E6=B5=8B=E6=96=B9=E6=B3=952?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Vision/tool/CameraHIK.py | 12 ++++++++++++
 get_position_test.py     |  6 ++++++
 2 files changed, 18 insertions(+)

diff --git a/Vision/tool/CameraHIK.py b/Vision/tool/CameraHIK.py
index 7212da5..6d8f0b9 100644
--- a/Vision/tool/CameraHIK.py
+++ b/Vision/tool/CameraHIK.py
@@ -39,6 +39,9 @@ class camera_HIK():
         frame = None
         if self.init_success==True:
             if portisopen(self.ip, self.port):
+                # ret, frame = self.cap.read()
+                # if ret == False:
+                self.reconnect_camera()
                 ret, frame = self.cap.read()
             else:
                 print('海康摄像头网络断开')
@@ -46,6 +49,10 @@ class camera_HIK():
             if portisopen(self.ip, self.port):
                 self.reconnect_camera()
                 ret, frame = self.cap.read()
+                # if ret == False:
+                #     self.init_success =False
+                # else:
+                #     self.init_success = True
             else:
                 print('海康摄像头网络断开')
 
@@ -55,6 +62,11 @@ class camera_HIK():
         if self.init_success == True:
             self.cap.release()
         self.cap = cv2.VideoCapture(self.camera_url)
+        ret, _ = self.cap.read()
+        if ret:
+            self.init_success = True
+        else:
+            self.init_success = False
         print("海康摄像头重连")
 
     def release_camera(self):
diff --git a/get_position_test.py b/get_position_test.py
index f1bda79..21669c2 100644
--- a/get_position_test.py
+++ b/get_position_test.py
@@ -5,6 +5,8 @@
 # @Author  : hjw
 # @File    : get_position_test.py
 '''
+import time
+
 from Vision.camera_coordinate_dete_img import Detection
 from Trace.handeye_calibration import *
 from Vision.tool.utils import get_disk_space
@@ -56,6 +58,10 @@ def detectionPerson_test():
         if img_src is not None:
             cv2.imshow('detectPerson', img_src)
             cv2.waitKey(1)
+            time.sleep(5)
+        else:
+            cv2.destroyAllWindows()
+
         print("person", person)
     # video_path = ''.join([os.getcwd(), '/Vision/model/data/1.mp4'])
     # cam = cv2.VideoCapture(video_path)