变频器集成以及增加点动控制(0209)

2026-02-09 11:36:37 +08:00
parent 88dfc53b9d
commit d6ad01274a
45 changed files with 7161 additions and 1578 deletions
--- a/vision/camera_picture.py
+++ b/vision/camera_picture.py
@ -0,0 +1,87 @@
+import os
+import threading
+from datetime import datetime
+import cv2
+import time
+
+def capture_camera_images(rtsp_url, save_dir, camera_name, img_count=10, interval=0.5, reverse=False):
+    """
+    从RTSP摄像头抓取指定数量图片并保存
+    :param rtsp_url: 摄像头RTSP地址
+    :param save_dir: 保存目录
+    :param camera_name: 摄像头名称（用于文件名区分）
+    :param img_count: 抓取图片数量
+    :param interval: 抓取间隔（秒）
+    :param reverse: 是否翻转180
+    """
+    # 创建摄像头专属保存目录
+    camera_save_dir = os.path.join(save_dir, camera_name)
+    if not os.path.exists(camera_save_dir):
+        os.makedirs(camera_save_dir)
+
+    # 抓取并保存图片
+    for i in range(img_count):
+        cap = None
+        try:
+            # 打开RTSP流
+            cap = cv2.VideoCapture(rtsp_url, cv2.CAP_FFMPEG)
+            if not cap.isOpened():
+                continue  
+            cap.set(cv2.CAP_PROP_BUFFERSIZE, 1)
+            ret, frame = cap.read()
+            if reverse: # 镜像翻转
+                frame = cv2.flip(frame, -1)
+            if ret and frame is not None:
+                img_filename = f"{camera_name}_{datetime.now().strftime('%Y%m%d%H%M%S')}_{i+1}.jpg"
+                img_path = os.path.join(camera_save_dir, img_filename)
+                cv2.imwrite(img_path, frame)
+                time.sleep(interval)
+        except Exception as e:
+            print(f"⚠ 摄像头 {camera_name} 第 {i+1} 次截图异常：{e}，跳过")
+        finally:
+            if cap is not None and cap.isOpened():
+                cap.release()
+    
+    print(f"✅ 摄像头 {camera_name} 完成，共保存 {img_count} 张浇筑满的图片")
+
+
+def save_camera_picture(save_dir="full_images", img_count=15, interval=1.5, camera_60_rtsp=None, camera_61_rtsp=None):
+    """
+    执行摄像头截图保存：
+    1. 创建以时间戳命名的目录，统一存放日志和图片
+    2. 抓取60和61摄像头各img_count张图片
+    :param save_dir: 保存浇筑满图片的顶级目录
+    :param img_count: 每次浇筑满, 保存的图片的总数量
+    :param interval: 保存两张浇筑满图片之间的时间间隔
+    """
+    # 0. 先创建上级目录save_dir（若不存在）
+    if not os.path.exists(save_dir):
+        os.makedirs(save_dir)
+
+    # 1. 创建保存浇筑满图片的根目录
+    root_dir = f"{save_dir}/PourFull_{datetime.now().strftime('%Y%m%d%H%M%S')}"
+    if not os.path.exists(root_dir):
+        os.makedirs(root_dir)
+
+    # 2. 抓取两个摄像头的图片
+    if camera_60_rtsp is None:
+        camera_60_rtsp = "rtsp://admin:XJ123456@192.168.250.60:554/streaming/channels/101"
+    
+    if camera_61_rtsp is None:
+        camera_61_rtsp = "rtsp://admin:XJ123456@192.168.250.61:554/streaming/channels/101"
+
+    # 并行抓取
+    t1 = threading.Thread(
+        target=capture_camera_images,
+        args=(camera_60_rtsp, root_dir, "camera60", img_count, interval, False),
+        daemon=True
+    )
+    t2 = threading.Thread(
+        target=capture_camera_images,
+        args=(camera_61_rtsp, root_dir, "camera61", img_count, interval, True),
+        daemon=True
+    )
+    t1.start()
+    t2.start()
+
+    return root_dir