pybaseutils 2.4.26__tar.gz → 2.4.33__tar.gz

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/PKG-INFO

@@ -1,19 +1,13 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.4.26
+Version: 2.4.33
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
 Author-email: 390737991@qq.com
 License: MIT
+Platform: UNKNOWN
 License-File: LICENCE
-Dynamic: author
-Dynamic: author-email
-Dynamic: description
-Dynamic: home-page
-Dynamic: license
-Dynamic: license-file
-Dynamic: summary
 
 base-utils
 ==========
@@ -148,3 +142,5 @@ base_utils(C++)
        DEBUG_IMSHOW("image", image);
        return 0;
    }
+
+

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils/__init__.py

@@ -8,4 +8,4 @@
 # --------------------------------------------------------
 """
 
-__version__ = '2.4.26'
+__version__ = '2.4.33'

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils/cvutils/camera.py

@@ -7,6 +7,7 @@ import subprocess
 class CameraCapture(object):
     def __init__(self, video: str or int = 0, fps=30, size=(1920, 1080), scale=1.0, input_format="mjpeg"):
         """
+        v4l2-ctl --device /dev/video0 --list-formats-ext # 查看摄像头在所有格式和分辨率下支持的帧率
         fuser /dev/video31  # 查看视频设备是否被占用
         查询视频设备分辨率： ffmpeg -f v4l2 -list_formats all -i /dev/video0
         常见的视频分辨率  ： 1920x1080 1280x720 640x480 352x288 320x240 176x144 160x120

pybaseutils-2.4.26/pybaseutils/cvutils/monitor.py → pybaseutils-2.4.33/pybaseutils/cvutils/frame_record.py

@@ -37,7 +37,7 @@ class Queue():
         return self.data[index]
 
 
-class StatusMonitor():
+class StatusRecord():
     """
     检测视频流状态,计算连续相邻两帧的相似性,
     判断当前状态{"运动":1,"静止":0},

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils/cvutils/video_utils.py

@@ -12,10 +12,11 @@ import cv2
 import numpy as np
 import math
 import time as timelib
+import subprocess
 from typing import Callable
 from tqdm import tqdm
 from pybaseutils import image_utils, file_utils
-from pybaseutils.cvutils import monitor
+from pybaseutils.cvutils import frame_record
 
 VIDEO_POSTFIX = ['*.mp4', '*.avi', '*.mov', "*.flv"]
 
@@ -52,6 +53,13 @@ def get_usb_camera(video=-1, max_index=50):
 
 
 def merge_video(video1, video2, output="./output.mp4"):
+    """
+    视频拼接
+    :param video1:
+    :param video2:
+    :param output:
+    :return:
+    """
     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}'
 
 
@@ -189,7 +197,7 @@ def video2frames_similarity(video_file, out_dir=None, func=None, prefix="", inte
     :param vis: 是否可视化显示
     :return:
     """
-    sm = monitor.StatusMonitor()
+    rec = frame_record.StatusRecord()
     name = os.path.basename(video_file).split(".")[0].replace("-", "_")
     if prefix: name = f"{prefix}_{name}"
     if not out_dir:  out_dir = os.path.join(os.path.dirname(video_file), name)
@@ -208,7 +216,7 @@ def video2frames_similarity(video_file, out_dir=None, func=None, prefix="", inte
             if func: curr_frame = func(curr_frame)
             if last_frame is None:
                 last_frame = curr_frame.copy()
-            diff = sm.get_frame_similarity(curr_frame, last_frame, size=(256, 256), vis=False)
+            diff = rec.get_frame_similarity(curr_frame, last_frame, size=(256, 256), vis=False)
             if diff > thresh:
                 frame_file = os.path.join(out_dir, "{}_{:0=6d}.jpg".format(name, count))
                 last_frame = curr_frame.copy()
@@ -333,13 +341,13 @@ def video_capture(video: int or str, save_video: str or int = None, interval=1,
         assert os.path.exists(video), f"video={video}"
     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]
+    clip_time = kwargs.get("clip_time", tuple())  # TODO 开始播放时间time[0]，结束播放时间time[1]，单位秒S
+    clip_index = kwargs.get("clip_index", tuple())  # TODO 开始播放位置clip[0]，结束播放位置clip[1]
     start, end = 0, -1
-    if time and fps > 0:
-        start, end = int(time[0] * fps), int(time[1] * fps)
-    elif clip:
-        start, end = int(clip[0]), int(clip[1])
+    if clip_time and fps > 0:
+        start, end = int(clip_time[0] * fps), int(clip_time[1] * fps)
+    elif clip_index:
+        start, end = int(clip_index[0]), int(clip_index[1])
     end = min(end, num_frames) if end > 0 else num_frames  # TODO 当num_frames<0时，使用0<end<count继续播放
     interval = fps if interval == -1 else interval  # 当interval=-1，表示interval=fps,即一秒一帧
     interval = int(fps / freq) if freq > 0 else interval
@@ -388,8 +396,8 @@ def video_iterator(video, save_video: str or int = None, interval=1, size=(), fr
     :param kwargs:回调函数输入参数,
                  delay: 控制显示延时,默认10S
                  title: 控制显示窗口名，默认video
-                 time: 开始播放时间time[0]，结束播放时间time[1]，单位秒S
-                 clip: 开始播放位置clip[0]，结束播放位置clip[1]
+                 clip_time: 开始播放时间time[0]，结束播放时间time[1]，单位秒S
+                 clip_index: 开始播放位置clip[0]，结束播放位置clip[1]
                  speed: 保存视频的播放速度，默认是播放帧率
                  save_fps: 保存视频的帧率，默认是播放帧率
     :return: frame, count, w, h, fps =data_info['frame'],data_info['count'],data_info['w'],data_info['h'],data_info['fps']
@@ -401,13 +409,13 @@ def video_iterator(video, save_video: str or int = None, interval=1, size=(), fr
         assert os.path.exists(video), f"video={video}"
     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]
+    clip_time = kwargs.get("clip_time", tuple())  # TODO 开始播放时间time[0]，结束播放时间time[1]，单位秒S
+    clip_index = kwargs.get("clip_index", tuple())  # TODO 开始播放位置clip[0]，结束播放位置clip[1]
     start, end = 0, -1
-    if time and fps > 0:
-        start, end = int(time[0] * fps), int(time[1] * fps)
-    elif clip:
-        start, end = int(clip[0]), int(clip[1])
+    if clip_time and fps > 0:
+        start, end = int(clip_time[0] * fps), int(clip_time[1] * fps)
+    elif clip_index:
+        start, end = int(clip_index[0]), int(clip_index[1])
     end = min(end, num_frames) if end > 0 else num_frames  # TODO 当num_frames<0时，使用0<end<count继续播放
     interval = fps if interval == -1 else interval  # 当interval=-1，表示interval=fps,即一秒一帧
     interval = int(fps / freq) if freq > 0 else interval
@@ -472,6 +480,47 @@ def rotation_task(frame, **kwargs):
     return frame
 
 
+def segment_video(video_file, save_file="", clip_time=(), clip_index=()):
+    """
+    分割视频
+    :param video_file:  视频文件
+    :param save_file:  输出分割视频文件
+    :param clip_time:  分割视频时间, clip_time[0]开始时间，结束时间clip_time[1]，单位秒S
+    :param clip_index: 分割视频索引, clip_index[0]开始位置，结束位置clip_index[1]
+    :return:
+    """
+    if not save_file:
+        name_id, postfix = file_utils.split_postfix(video_file)
+        save_file = os.path.join(os.path.dirname(video_file), f"{name_id}_crop.mp4")
+    os.makedirs(os.path.dirname(save_file), exist_ok=True)
+    assert os.path.exists(video_file), f"输入视频文件不存在: {video_file}"
+    cmd = ["ffmpeg", "-i", video_file]
+    if clip_time and len(clip_time) == 2:
+        start, end = clip_time
+        # -ss 和 -to 放在 -i 之后实现精确解码切割（无 seek 误差）
+        cmd.extend(["-ss", str(start), "-to", str(end)])
+        # 若不需重新编码且可接受关键帧对齐，可启用下行加速：
+        # cmd.extend(["-c", "copy"])
+    elif clip_index and len(clip_index) == 2:
+        c0, c1 = int(clip_index[0]), int(clip_index[1])
+        # select 滤镜选择帧号范围内的帧，并重置时间戳
+        vf_filter = f"select='between(n,{c0},{c1})',setpts=N/FRAME_RATE/TB"
+        af_filter = f"aselect='between(n,{c0},{c1})',asetpts=N/SR/TB"
+        cmd.extend(["-vf", vf_filter, "-af", af_filter, "-vsync", "vfr"])
+    else:
+        raise ValueError("必须提供 clip_time 或 clip_index 且参数为二元组")
+    # 覆盖输出文件
+    cmd.extend(["-y", save_file])
+    # 执行命令
+    try:
+        print(" ".join(cmd))
+        subprocess.run(cmd, check=True, capture_output=True, text=True)
+        print(f"视频分割完成: {save_file}")
+    except subprocess.CalledProcessError as e:
+        print(f"ffmpeg 执行失败:\n{e.stderr}")
+        raise
+
+
 if __name__ == "__main__":
     # video_file = "/home/dm/nasdata/release/CSDN/双目测距Demo视频(Python).MP4"
     # video_file = "/home/dm/视频/双目测距Demo视频(Python).mp4"

pybaseutils-2.4.33/pybaseutils/dict_uils.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-08-20 16:04:27
+# @Brief  :
+# --------------------------------------------------------
+"""
+from pybaseutils.dict_utils import *

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils/file_utils.py

@@ -1607,6 +1607,49 @@ def zip_file(src, dst=None, s=None):
     return dst
 
 
+def json2jsonl(json_file: str, jsonl_out="./data.jsonl"):
+    """
+    将json转换为jsonl
+    :param json_file: json文件目录或json文件路径
+    :param jsonl_out:
+    :return:
+    """
+    data_list = []
+    if json_file.endswith(".json"):
+        with open(json_file, 'r', encoding='utf-8') as f:
+            data_list = json.load(f)  # 假设 data 是一个列表
+    elif os.path.isdir(json_file):
+        files = glob.glob(os.path.join(json_file, "*.json"))
+        for file in files:
+            with open(file, 'r', encoding='utf-8') as f:
+                data = json.load(f)  # 假设 data 是一个字典
+            data_list.append(data)
+    with open(jsonl_out, 'w', encoding='utf-8') as f:
+        for item in data_list:
+            f.write(json.dumps(item, ensure_ascii=False) + '\n')
+    print(f"转换成功，保存文件：{jsonl_out} ,共{len(data_list)}条记录")
+    return data_list
+
+
+def jsonl2json(jsonl_file: str, json_out="./data.json"):
+    """
+    将 JSONL 文件转换为 JSON 文件
+    :param jsonl_file: JSONL 文件路径
+    :param json_out: 输出的 JSON 文件路径
+    :return: 包含所有数据的列表
+    """
+    data_list = []
+    with open(jsonl_file, 'r', encoding='utf-8') as f:
+        for line in f:
+            line = line.strip()
+            if line:  # 跳过空行
+                data_list.append(json.loads(line))
+    with open(json_out, 'w', encoding='utf-8') as f:
+        json.dump(data_list, f, ensure_ascii=False, indent=2)
+    print(f"转换成功，保存文件：{json_out}，共 {len(data_list)} 条记录")
+    return data_list
+
+
 if __name__ == '__main__':
     from pybaseutils import time_utils
 

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils/json_utils.py

@@ -15,7 +15,7 @@ import PIL.Image as Image
 from collections import Counter
 from typing import List, Tuple, Dict
 from pybaseutils.file_utils import load_json, read_json_data, save_json, write_json_path
-from pybaseutils.dict_uils import *
+from pybaseutils.dict_utils import *
 
 
 def get_most_common(data: list, topK=None):
@@ -163,8 +163,9 @@ def get_index(data: list, v):
     return indices
 
 
+
 if __name__ == "__main__":
     # data = {'C': 0, 'A': 5, 'B': 3, 'D': 2}
     # print(dict_sort(data))
-    i= get_index(data=[], v=7)
+    i = get_index(data=[], v=7)
     print(i)

pybaseutils-2.4.33/pybaseutils/log.py

@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2022-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
+"""
+import os
+import sys
+import datetime
+from loguru import logger
+
+LOG_FORMAT = {
+    "simple":   "<level>{extra[time]}|{level:7}| {message}</level>",
+    "name":     "<level>{extra[time]}|{level:7}|{name} {line}| {message}</level>",  # 打印文件名
+    "module":   "<level>{extra[time]}|{level:7}|{module} {line}| {message}</level>",  # 打印模块名
+    "function": "<level>{extra[time]}|{level:7}|{module}.{function} {line}| {message}</level>",
+    "precise":  "<level>{extra[time]}|{level:7}|{module}.{function} {line}| {message}</level>",
+    "all":      "<level>{extra[time]}|{level:7}|{name}.{module}.{function} {line}| {message}</level>",# 打印函数
+}
+
+
+def set_logger(name=None, level="debug", logfile=None, format="simple", is_main_process=True,
+               rotation="1 days", retention="3 days"):
+    """
+    logger = set_logger(level="debug", logfile="log.txt")
+    url: https://www.cnblogs.com/shiyitongxue/p/17870527.html
+    :param level: 设置log输出级别:debug,info,warning,error
+    :param logfile: log保存路径，如果为None，则在控制台打印log
+    :param is_main_process: 是否是主进程
+    :param rotation: 日志文件轮转策略,时间或者大小，默认1天
+                    rotation="1 minutes"  # 每分钟轮转
+                    rotation="1 hours"    # 每小时轮转
+                    rotation="1 days"     # 每天轮转
+                    rotation="1 weeks"    # 每周轮转
+                    rotation="10 MB"      # 文件达到10MB时轮转
+                    rotation="500 MB"     # 文件达到500MB时轮转
+    :param retention: 日志文件保留时长，默认3天
+                    retention="7 days"    # 保留7天
+                    retention="1 months"   # 保留1个月
+                    retention="20 GB"     # 保留最近20GB的日志（基于大小）
+    :return:
+    """
+
+    def call_time(record):
+        # 显式固化 logger 调用时刻，避免异步 sink 输出时产生“当前打印时间”的误解
+        if format == "precise":
+            record["extra"]["time"] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
+        else:
+            record["extra"]["time"] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    f = LOG_FORMAT.get(format, LOG_FORMAT.get("simple"))
+    logger.remove()  # 去除默认的LOG，避免重复打印
+    logger_ = logger.patch(call_time)
+    if is_main_process:
+        if logfile: logger_.add(logfile, level=level.upper(), rotation=rotation, retention=retention, format=f,
+                                enqueue=True,  # 异步写入，会重新打开文件
+                                watch=True,  # 避免误删日志文件
+                                catch=True,
+                                )
+        logger_.add(sys.stderr, level=level.upper(), format=f)
+    else:
+        logger_.add(sys.stderr, level="ERROR", format=f)
+    return logger_
+
+
+def get_logger():
+    return logger
+
+
+if __name__ == '__main__':
+    import traceback
+    import time
+
+    logfile = "./log.log"
+    logger = set_logger(name="demo", is_main_process=True, format="precise", level="debug", logfile=logfile)
+    # logger = set_logger(name="demo", is_main_process=True, format="module", level="debug", logfile=logfile)
+
+    for i in range(1000):
+        try:
+            t = time.time()
+            date = datetime.datetime.fromtimestamp(t).strftime("%H:%M:%S.%f")[:-3]
+            logger.debug(f"{date} debug {i}")
+            # logger.info(f"info {i}")
+            # logger.warning(f"warning {i}")
+            # logger.error(f"error {i}")
+            time.sleep(0.01)
+        except Exception as e:
+            e = traceback.format_exc()
+            logger.error(e)

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils.egg-info/PKG-INFO

@@ -1,19 +1,13 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.4.26
+Version: 2.4.33
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
 Author-email: 390737991@qq.com
 License: MIT
+Platform: UNKNOWN
 License-File: LICENCE
-Dynamic: author
-Dynamic: author-email
-Dynamic: description
-Dynamic: home-page
-Dynamic: license
-Dynamic: license-file
-Dynamic: summary
 
 base-utils
 ==========
@@ -148,3 +142,5 @@ base_utils(C++)
        DEBUG_IMSHOW("image", image);
        return 0;
    }
+
+

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/pybaseutils.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ pybaseutils/config_utils.py
 pybaseutils/coords_utils.py
 pybaseutils/data_utils.py
 pybaseutils/dict_uils.py
+pybaseutils/dict_utils.py
 pybaseutils/file_utils.py
 pybaseutils/font_utils.py
 pybaseutils/geometry_tools.py
@@ -76,7 +77,7 @@ pybaseutils/converter/prelabelme.py
 pybaseutils/cvutils/__init__.py
 pybaseutils/cvutils/camera.py
 pybaseutils/cvutils/corner_utils.py
-pybaseutils/cvutils/monitor.py
+pybaseutils/cvutils/frame_record.py
 pybaseutils/cvutils/mouse_utils.py
 pybaseutils/cvutils/nms_utils.py
 pybaseutils/cvutils/video_utils.py
@@ -186,6 +187,7 @@ test_py/demo_standard_video .py
 test_py/demo_taichi.py
 test_py/demo_video2frame.py
 test_py/demo_video_crop.py
+test_py/demo_video_segm.py
 test_py/demo_voc_crop.py
 test_py/demo_voc_vis.py
 test_py/demo_word_similar.py
@@ -203,6 +205,7 @@ test_py/test_camera.py
 test_py/Image_enhance/__init__.py
 test_py/Image_enhance/dmeo01.py
 test_py/WebCrawler/__init__.py
+test_py/WebCrawler/download_pdf.py
 test_py/WebCrawler/search_baidu.py
 test_py/WebCrawler/search_biying.py
 test_py/aije/__init__.py
@@ -248,6 +251,7 @@ test_py/converter/TT100K.py
 test_py/converter/WaterMeters1.py
 test_py/converter/WaterMeters2.py
 test_py/converter/__init__.py
+test_py/converter/car_accident.py
 test_py/converter/concat_coco.py
 test_py/converter/convert_coco2voc.py
 test_py/converter/convert_cvat2labelme.py

pybaseutils-2.4.33/test_py/WebCrawler/download_pdf.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2026-06-24 15:32:23
+# @Brief  :
+# --------------------------------------------------------
+"""

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/test_py/aije/demo_video_aije.py

@@ -10,7 +10,7 @@ import cv2
 import numpy as np
 from tqdm import tqdm
 from pybaseutils import file_utils, image_utils
-from pybaseutils.cvutils import video_utils, monitor
+from pybaseutils.cvutils import video_utils, frame_record
 
 # thresh_dict = {"1号视角": 0.5, "2号视角": 0.15, "3号视角": 0.3, "4号视角": 0.3}
 # thresh_dict = {"一号位": 0.58, "平视": 0.35, "俯视": 0.35, "右视": 0.25, "左视": 0.3}

pybaseutils-2.4.33/test_py/converter/car_accident.py

@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2026-06-23 17:33:01
+# @Brief  :
+# --------------------------------------------------------
+"""
+import os
+import cv2
+from collections import defaultdict
+from pybaseutils import json_utils, pandas_utils, image_utils, file_utils, dict_utils
+from pybaseutils.cvutils import video_utils
+
+class_maps = {
+    "before crush": {"color": (255, 0, 0), "name": "crush0", "shift": [0, -1.0]},
+    "crush moment": {"color": (0, 255, 0), "name": "crush1", "shift": [-1.0, 1.0]},
+    "after crush": {"color": (0, 0, 255), "name": "crush2", "shift": [-1.0, 0]},
+}
+
+
+def parser_video(video_file, csv_file, vis=True):
+    print("video_file", video_file)
+    print(" csv_file", csv_file)
+    pd_data = pandas_utils.read_csv(csv_file, sep=",")
+    pd_dict = pandas_utils.df2dict(pd_data, orient="index")
+    label_info = defaultdict(list)
+    video_cap = video_utils.get_video_capture(video_file)
+    width, height, num_frames, fps = video_utils.get_video_info(video_cap)
+    for info in pd_dict.values():
+        index = info['frame_num']
+        label = info['type'].strip()
+        label_info[label].append(index)
+        if not vis: continue
+        # 设置抽帧的位置
+        video_cap.set(cv2.CAP_PROP_POS_FRAMES, index)
+        ret, frame = video_cap.read()
+        if not ret: break
+        color = class_maps[label]["color"]
+        text = "{:3d}-{}".format(index, label)
+        frame = image_utils.draw_text(frame, (0, 0), text, color=color, fontScale=1, thickness=2)
+        image_utils.show_image("image", frame, delay=20)
+    return label_info, fps, num_frames
+
+
+def segment_video_dir(video_dir, csv_dir, output="", vis=False):
+    video_files = file_utils.get_files_list(video_dir, postfix=file_utils.VIDEO_POSTFIX)
+    for i, video_file in enumerate(video_files):
+        if not output: output = os.path.dirname(video_file)
+        name, ext = os.path.splitext(os.path.basename(video_file))
+        csv_file = os.path.join(csv_dir, name.replace("-", "_") + ".csv")
+        if not os.path.exists(csv_file):
+            print("not exists:{}".format(csv_file))
+            continue
+        label_info, fps, num_frames = parser_video(video_file, csv_file, vis=vis)
+        for label, indices in label_info.items():
+            clip = [min(indices), max(indices)]
+            name = class_maps[label]["name"]
+            shift = class_maps[label]["shift"]
+            clip_index = [clip[0] + int(shift[0] * fps), clip[1] + int(shift[1] * fps)]
+            clip_index = [max(0, clip_index[0]), min(num_frames, clip_index[1])]
+            if (clip_index[1] - clip_index[0]) / fps < 0.5: continue
+            nid, postfix = file_utils.split_postfix(video_file)
+            # save_file = os.path.join(output, f"{nid}_{name}_crop{clip[0]}_{clip[1]}.mp4")
+            save_file = os.path.join(output, name, f"video{i:03d}_{name}.mp4")
+            video_utils.segment_video(video_file, save_file=save_file, clip_index=clip_index)
+
+
+if __name__ == '__main__':
+    video_dir = '/media/PKing/dev1/project/Crash dataset/video/'
+    csv_dir = "/media/PKing/dev1/project/Crash dataset/label/csv"
+    output = '/media/PKing/dev1/project/Crash dataset/video-crop/'
+    segment_video_dir(video_dir, csv_dir, output=output)

{pybaseutils-2.4.26 → pybaseutils-2.4.33}/test_py/demo_rename.py

@@ -23,6 +23,6 @@ def rename_files(src, dst, prefix="", postfix=None, remove=False):
 
 
 if __name__ == '__main__':
-    image_dir = "/home/PKing/edudata/dataset/AIJE/【TOP】技能人才系统_数据集管理/05-东莞基地/东莞基地-2025-08-19/images"
+    image_dir = "/home/PKing/nasdata/tmp/tmp/car-accident/dataset-v1"
     output = image_dir
-    rename_files(image_dir, output, prefix="dg_image", remove=True)
+    rename_files(image_dir, output, prefix="video", remove=True)

pybaseutils-2.4.33/test_py/demo_video_segm.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+"""
+    @Author : PKing
+    @E-mail : 390737991@qq.com
+    @Date   : 2022-12-31 11:37:30
+    @Brief  :
+"""
+import os
+import cv2
+import numpy as np
+from pybaseutils import file_utils, image_utils
+from pybaseutils.cvutils import video_utils
+
+
+def task(frame, **kwargs):
+    frame = image_utils.resize_image(frame, size=(1280, None))
+    return frame
+
+
+if __name__ == "__main__":
+    video_file = "/home/PKing/Pictures/ailt-video.mp4"
+    width, height, num_frames, fps = video_utils.get_video_info(video_file)
+    clip_time = (13.5, 19.0)
+    clip_index = (clip_time[0] * fps, clip_time[1] * fps)
+    # video_utils.segment_video(video_file, clip_time=clip_time)
+    video_utils.segment_video(video_file, clip_index=clip_index)