feat: 增加循跡回歸雙服務端

ocr_server/ocr_infer_server.py

@@ -5,6 +5,7 @@ import cv2
 import numpy as np
 import requests  
 import base64
+import datetime 
 
 
 
@@ -58,6 +59,7 @@ def ocr_api_request(image_base64):
 
 if __name__ == "__main__":
 
+    logger.info("ocr server 开始加载")
     cfg = toml.load('/home/evan/Workplace/project_infer/cfg_infer_server.toml')
 
     # 配置日志输出
@@ -68,11 +70,11 @@ if __name__ == "__main__":
     # camera_socket.connect(f"tcp://localhost:{cfg['camera']['camera2_port']}")
     # logger.info("connect camera success")
 
-    cap = cv2.VideoCapture(20)
-    cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter.fourcc('M','J','P','G'))
-    cap.set(cv2.CAP_PROP_FPS, 20)
-    cap.set(cv2.CAP_PROP_FRAME_WIDTH, 960)
-    cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 540)
+    # cap = cv2.VideoCapture(20)
+    # cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter.fourcc('M','J','P','G'))
+    # cap.set(cv2.CAP_PROP_FPS, 20)
+    # cap.set(cv2.CAP_PROP_FRAME_WIDTH, 960)
+    # cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 540)
 
     # 初始化 server
     context = zmq.Context()
@@ -83,7 +85,7 @@ if __name__ == "__main__":
     import signal
     import sys
     def signal_handler(signum, frame):
-        logger.info("Received signal, exiting...")
+        logger.info(f"接收到退出信号 {signum}, 退出中")
         socket.close()
         context.term()
         sys.exit(0)
@@ -92,15 +94,30 @@ if __name__ == "__main__":
     signal.signal(signal.SIGTERM, signal_handler)
     signal.signal(signal.SIGINT, signal_handler)
 
+    logger.info("ocr server 加载完成")
+
     while True:
         message1 = socket.recv_string()
-        logger.info("recv client request")
-        ret, frame = cap.read()
+        logger.info("收到客户端请求")
+
+        logger.info("构造摄像头")
+        cap = cv2.VideoCapture(20)
+        cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter.fourcc('M','J','P','G'))
+        cap.set(cv2.CAP_PROP_FPS, 20)
+        cap.set(cv2.CAP_PROP_FRAME_WIDTH, 960)
+        cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 540)
+
+        for _ in range(5):
+            ret, frame = cap.read()
+            cv2.waitKey(1)
+
         if ret:
             frame = frame[:,0:480]
             frame = cv2.rotate(frame, cv2.ROTATE_90_COUNTERCLOCKWISE)
             
-            output_file_path = 'rotate.jpg'
+            now = datetime.datetime.now()  
+            timestamp = now.strftime("%Y%m%d_%H%M%S")  
+            output_file_path = f"./saved_picture/{timestamp}.jpg"  
             success = cv2.imwrite(output_file_path, frame)
             
             _, frame = cv2.imencode('.jpg', frame)
@@ -108,13 +125,19 @@ if __name__ == "__main__":
             encoded_image = base64.b64encode(frame).decode('utf-8')
             
             result = ocr_api_request(encoded_image)
-            print(result)
+            # print(result)
             if result != None:
                 socket.send_pyobj({'code': 0, 'content': result.get('words_result')})
+                logger.info(f"ocr 返回 {result.get('words_result')}")
             else:
                 socket.send_pyobj({'code': -1, 'content': " ocr 没找到文字"})
+                logger.error("ocr 没找到文字")
         else:
             socket.send_pyobj({'code': -1, 'content': "ocr 摄像头读取出错"})
+            logger.critical("ocr 摄像头读取出错")
+
+        cap.release()
+        
         if cv2.waitKey(1) == 27:
             break
     logger.info("ocr infer server exit")