87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
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 |