pybaseutils 2.4.8__tar.gz → 2.4.9__tar.gz

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybaseutils
-Version: 2.4.8
+Version: 2.4.9
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/pybaseutils/__init__.py

@@ -8,4 +8,4 @@
 # --------------------------------------------------------
 """
 
-__version__ = '2.4.8'
+__version__ = '2.4.9'

pybaseutils-2.4.9/pybaseutils/cvutils/camera.py

@@ -0,0 +1,90 @@
+import cv2
+import time
+import numpy as np
+import subprocess
+
+
+class CameraCapture(object):
+    def __init__(self, video: str or int = 0, size=(1920, 1080), fps=30):
+        """
+        :param video: 视频设备路径或索引（如 0 或 "/dev/video0"）
+        :param size: 视频分辨率 (宽, 高)
+        :param fps: 视频帧率
+        """
+        self.size = size
+        self.fps = fps
+        self.stopped = False
+        if isinstance(video, int): video = f"/dev/video{video}"
+        # 构建 FFmpeg 命令
+        # -re: 以原生帧率读取（模拟直播流）
+        # -fflags nobuffer: 关键！禁用缓冲区
+        # -flags low_delay: 关键！低延迟模式
+        # -probesize 32: 减小探测包大小，加快启动
+        # -pix_fmt bgr24: 直接输出 BGR 格式，方便 OpenCV/Numpy 使用，避免后续转换
+        command = [
+            'ffmpeg',
+            '-re',  # 按帧率读取
+            '-fflags', 'nobuffer',  # 无缓冲
+            '-flags', 'low_delay',  # 低延迟
+            '-probesize', '32',  # 快速探测
+            '-i', video,  # 输入设备
+            '-f', 'rawvideo',  # 输出原始视频流
+            '-pix_fmt', 'bgr24',  # 像素格式 BGR (OpenCV 格式)
+            '-s', f'{size[0]}x{size[1]}',  # 分辨率
+            '-r', str(fps),  # 帧率
+            '-'  # 输出到 stdout
+        ]
+        # 启动进程
+        self.pipe = subprocess.Popen(command, stdout=subprocess.PIPE, bufsize=10 ** 8)
+        print(f"command: {' '.join(command)}")
+
+    def read(self):
+        if self.stopped:
+            self.stop()
+            return False, None
+        # 读取一帧的数据量 (宽 * 高 * 3通道)
+        buf = self.pipe.stdout.read(self.size[0] * self.size[1] * 3)
+        if len(buf) != self.size[0] * self.size[1] * 3:
+            self.stop()
+            return False, None  # 读取失败或结束
+        # 将字节流转换为 numpy 数组
+        frame = np.frombuffer(buf, dtype=np.uint8).reshape((self.size[1], self.size[0], 3))
+        return True, frame
+
+    def release(self):
+        self.stop()
+
+    def stop(self):
+        self.stopped = True
+        time.sleep(0.1)
+        self.pipe.terminate()  # 终止进程
+        try:
+            outs, errs = self.pipe.communicate(timeout=2)
+        except subprocess.TimeoutExpired as err:
+            # 如果 communicate 也超时（极少见），强制杀死
+            self.pipe.kill()
+            self.pipe.communicate()  # 再次尝试清理
+            print(err)
+
+    def display(self):
+        while True:
+            t1 = time.time()
+            ret, frame = self.read()
+            t2 = time.time()
+            if not ret:
+                break
+            t21 = (t2 - t1) * 1000
+            print(f"image shape: {frame.shape},耗时: {t21:.3f}ms")
+            cv2.imshow('FFmpeg Low Latency', frame)
+            if cv2.waitKey(30) & 0xFF == ord('q'):
+                break
+
+
+if __name__ == '__main__':
+    fps = 30
+    width = 1920
+    height = 1080
+    video = 0  # Windows 下可能是 0 或 "video=Integrated Webcam"
+    # video = "/home/PKing/Videos/demo-src.mp4"  # Windows 下可能是 0 或 "video=Integrated Webcam"
+    cap = CameraCapture(video=video, size=(width, height), fps=fps)
+    cap.display()

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/pybaseutils/cvutils/video_utils.py

@@ -24,6 +24,26 @@ get_video_info = image_utils.get_video_info
 get_video_writer = image_utils.get_video_writer
 
 
+def get_usb_camera(video=-1, max_index=50):
+    """
+    获取当前系统中连接的USB摄像头索引。
+    :param video: 视频路径或摄像头索引,默认值为-1表示自动检测。当video>=0时,不进行检测，直接返回video。
+    :param max_index: 最大检测索引范围,默认值为50。
+    :return: 一摄像头索引,camera=-1表示未检测到摄像头。
+    """
+    if video >= 0: return video
+    print("正在检测USB摄像头ID...")
+    camera = -1
+    for i in range(max_index):
+        cap = cv2.VideoCapture(i)
+        if cap.isOpened():
+            camera = i
+            cap.release()
+            break
+    print("检测到USB摄像头ID={}".format(camera))
+    return camera
+
+
 def merge_video(video1, video2, output="./output.mp4"):
     cmd = f'ffmpeg -i {video1} -i {video2} -filter_complex "[0:v:0][1:v:0]concat=n=2:v=1:a=0[outv]" -map "[outv]" {output}'
 
@@ -304,8 +324,8 @@ def video_capture(video: int or str, save_video: str or int = None, interval=1,
     video = file_utils.str2number(video)
     if isinstance(video, str) and os.path.isfile(video):
         assert os.path.exists(video), f"video={video}"
-    video_cap = image_utils.get_video_capture(video, fps=None)
-    w, h, num_frames, fps = image_utils.get_video_info(video_cap, **kwargs)
+    cap = image_utils.get_video_capture(video, fps=None)
+    w, h, num_frames, fps = image_utils.get_video_info(cap, **kwargs)
     time = kwargs.get("time", tuple())  # TODO 开始播放时间time[0]，结束播放时间time[1]，单位秒S
     clip = kwargs.get("clip", tuple())  # TODO 开始播放位置clip[0]，结束播放位置clip[1]
     start, end = 0, -1
@@ -321,7 +341,7 @@ def video_capture(video: int or str, save_video: str or int = None, interval=1,
     count = 0
     video_writer = None
     while True:
-        ret, frame = video_cap.read()
+        ret, frame = cap.read()
         if count % interval == 0 and count >= start:
             # TODO 设置抽帧的位置，但某些格式视频容易出现问题
             # if isinstance(video, str): video_cap.set(cv2.CAP_PROP_POS_FRAMES, count)
@@ -335,7 +355,7 @@ def video_capture(video: int or str, save_video: str or int = None, interval=1,
                 if not video_writer: video_writer = image_utils.get_video_writer(save_video, w, h, save_fps)
                 video_writer.write(frame)
         count += 1
-    video_cap.release()
+    cap.release()
     if video_writer:
         print("save video:{}".format(save_video))
         video_writer.release()
@@ -347,8 +367,8 @@ def video_iterator(video, save_video: str or int = None, interval=1, size=(), fr
     读取摄像头或者视频流迭代器
     Usage:
         from pybaseutils.cvutils import video_utils
-        video_cap = video_utils.video_iterator(video, save_video, time=(4, 10))
-        for data_info in video_cap:
+        cap = video_utils.video_iterator(video, save_video, time=(4, 10))
+        for data_info in cap:
             frame = data_info["frame"]
             ...
     :param video: String 视频文件，如*.avi,*.mp4,...
@@ -372,8 +392,8 @@ def video_iterator(video, save_video: str or int = None, interval=1, size=(), fr
     video = file_utils.str2number(video)
     if isinstance(video, str) and os.path.isfile(video):
         assert os.path.exists(video), f"video={video}"
-    video_cap = image_utils.get_video_capture(video, fps=None)
-    w, h, num_frames, fps = image_utils.get_video_info(video_cap, **kwargs)
+    cap = image_utils.get_video_capture(video, fps=None)
+    w, h, num_frames, fps = image_utils.get_video_info(cap, **kwargs)
     time = kwargs.get("time", tuple())  # TODO 开始播放时间time[0]，结束播放时间time[1]，单位秒S
     clip = kwargs.get("clip", tuple())  # TODO 开始播放位置clip[0]，结束播放位置clip[1]
     start, end = 0, -1
@@ -393,16 +413,16 @@ def video_iterator(video, save_video: str or int = None, interval=1, size=(), fr
     t0 = -1
     while True:
         systime = timelib.time()
-        ret, frame = (False, None) if use_fast else video_cap.read()  # 读取视频很快，但读取USB摄像头比较耗时
-        t = video_cap.get(cv2.CAP_PROP_POS_MSEC) / 1000  # 获得视频偏移量毫秒为单位
+        ret, frame = (False, None) if use_fast else cap.read()  # 读取视频很快，但读取USB摄像头比较耗时
+        t = cap.get(cv2.CAP_PROP_POS_MSEC) / 1000  # 获得视频偏移量毫秒为单位
         if t0 < 0 < t: t0 = t
         if t0 > 0: t = t - t0  # 获得视频偏移量毫秒为单位
         finish = 0 < end <= (count + interval)
         if count % interval == 0 and count >= start:
             # TODO 设置抽帧的位置，但某些格式视频容易出现问题
             if use_fast and isinstance(video, str):
-                video_cap.set(cv2.CAP_PROP_POS_FRAMES, count)
-                ret, frame = video_cap.read()
+                cap.set(cv2.CAP_PROP_POS_FRAMES, count)
+                ret, frame = cap.read()
             if not ret or 0 < end <= count or frame is None: break
             if size: frame = image_utils.resize_image(frame, size=size)
             if task: frame = task(frame, **kwargs)
@@ -421,7 +441,7 @@ def video_iterator(video, save_video: str or int = None, interval=1, size=(), fr
                 video_writer.write(frame)
             if data_info['finish']: break  # 控制结束
         count += 1
-    video_cap.release()
+    cap.release()
     if video_writer:
         print("save video:{}".format(save_video))
         video_writer.release()

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/pybaseutils/file_utils.py

@@ -1579,6 +1579,7 @@ def zip_file(src, dst=None, s=None):
     return dst
 
 
+
 if __name__ == '__main__':
     from pybaseutils import time_utils
 

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/pybaseutils/image_utils.py

@@ -3293,33 +3293,34 @@ def get_video_capture(video, width=None, height=None, fps=None):
     :param fps:  设置视频播放帧率
     :return:
     """
-    video_cap = cv2.VideoCapture(video)
+    cap = cv2.VideoCapture(video)
     # 设置分辨率
     if width:
-        video_cap.set(cv2.CAP_PROP_FRAME_WIDTH, width)
+        cap.set(cv2.CAP_PROP_FRAME_WIDTH, width)
     if height:
-        video_cap.set(cv2.CAP_PROP_FRAME_HEIGHT, height)
+        cap.set(cv2.CAP_PROP_FRAME_HEIGHT, height)
     if fps:
-        video_cap.set(cv2.CAP_PROP_FPS, fps)
-    return video_cap
+        cap.set(cv2.CAP_PROP_FPS, fps)
+    # cap.set(cv2.CAP_PROP_BUFFERSIZE, 1)  # 设置缓冲区大小为1，确保每次读取的都是最新，避免旧帧积压导致的延迟。
+    return cap
 
 
-def get_video_info(video_cap, disp=True, **kwargs):
+def get_video_info(cap, disp=True, **kwargs):
     """
     获得视频的基础信息
-    :param video_cap:视频对象 或者视频文件路径 int | str | cv2.VideoCapture
+    :param cap:视频对象 或者视频文件路径 int | str | cv2.VideoCapture
     :param disp: 是否显示视频信息
     :return:
     """
-    isfile = isinstance(video_cap, str) or isinstance(video_cap, int)
-    if isfile: video_cap = get_video_capture(video_cap)
-    width = int(video_cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-    height = int(video_cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    num_frames = int(video_cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    fps = video_cap.get(cv2.CAP_PROP_FPS)
+    isfile = isinstance(cap, str) or isinstance(cap, int)
+    if isfile: cap = get_video_capture(cap)
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    num_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    fps = cap.get(cv2.CAP_PROP_FPS)
     fps = math.ceil(fps)
     if disp: print("read video:width:{},height:{},fps:{},num_frames:{}".format(width, height, fps, num_frames))
-    if isfile: video_cap.release()
+    if isfile: cap.release()
     return width, height, num_frames, fps
 
 

pybaseutils-2.4.9/pybaseutils/nets_utils.py

@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2026-03-02 11:26:43
+# @Brief  :
+# --------------------------------------------------------
+"""
+import cv2
+import socket
+
+
+def get_local_ip():
+    """获取本机真实局域网IP地址"""
+    try:
+        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        s.connect(("8.8.8.8", 80))
+        ip = s.getsockname()[0]
+        s.close()
+        return ip
+    except Exception:
+        return "127.0.0.1"
+

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/pybaseutils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybaseutils
-Version: 2.4.8
+Version: 2.4.9
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan

{pybaseutils-2.4.8 → pybaseutils-2.4.9}/pybaseutils.egg-info/SOURCES.txt

@@ -18,6 +18,7 @@ pybaseutils/image_utils.py
 pybaseutils/json_utils.py
 pybaseutils/log.py
 pybaseutils/log_utils.py
+pybaseutils/nets_utils.py
 pybaseutils/numpy_utils.py
 pybaseutils/pandas_utils.py
 pybaseutils/plot_utils.py
@@ -73,6 +74,7 @@ pybaseutils/converter/convert_yolo2labelme.py
 pybaseutils/converter/convert_yolo2voc.py
 pybaseutils/converter/prelabelme.py
 pybaseutils/cvutils/__init__.py
+pybaseutils/cvutils/camera.py
 pybaseutils/cvutils/corner_utils.py
 pybaseutils/cvutils/monitor.py
 pybaseutils/cvutils/mouse_utils.py
@@ -196,6 +198,7 @@ test_py/men_tracemalloc.py
 test_py/model_process.py
 test_py/performance.py
 test_py/slowfastnet.py
+test_py/test_camera.py
 test_py/Image_enhance/__init__.py
 test_py/Image_enhance/dmeo01.py
 test_py/WebCrawler/__init__.py

pybaseutils-2.4.9/test_py/test_camera.py

@@ -0,0 +1,28 @@
+import numpy as np
+import cv2
+import time
+from pybaseutils.cvutils.camera import CameraCapture
+
+
+def display(cap: CameraCapture):
+    while True:
+        t1 = time.time()
+        ret, frame = cap.read()
+        t2 = time.time()
+        if not ret:
+            break
+        t21 = (t2 - t1) * 1000
+        print(f"image shape: {frame.shape},耗时: {t21:.3f}ms")
+        cv2.imshow('FFmpeg Low Latency', frame)
+        if cv2.waitKey(30) & 0xFF == ord('q'):
+            break
+
+
+if __name__ == '__main__':
+    fps = 5
+    width = 1920
+    height = 1080
+    # video = 0  # Windows 下可能是 0 或 "video=Integrated Webcam"
+    video = "/home/PKing/Videos/demo-src.mp4"  # Windows 下可能是 0 或 "video=Integrated Webcam"
+    cap = CameraCapture(video=video, size=(width, height), fps=fps)
+    display(cap)