pybaseutils 2.2.19__tar.gz → 2.2.21__tar.gz

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pybaseutils
-Version: 2.2.19
+Version: 2.2.21
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
@@ -36,11 +36,15 @@ pybaseutils(Python)
 
 pybaseutils是个人开发常用的python库，集成了python等常用的算法
 
--  安装方法1：pip install
--  pybaseutils (有延时，可能不是最新版本)
+-  安装方法1：pip install –upgrade -i
+   https://pypi.tuna.tsinghua.edu.cn/simple pybaseutils
+   (有延时，可能不是最新版本)
 -  安装方法2：pip install –upgrade -i https://pypi.org/simple
    pybaseutils (从pypi源下载最新版本)
 
+(设置默认的镜像源：pip config set global.index-url
+https://pypi.tuna.tsinghua.edu.cn/simple)
+
 一些问题修复说明
 ----------------
 

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/README.md

@@ -18,10 +18,10 @@ base_util是个人开发常用的C++库，集成了C/C++ OpenCV等常用的算
 
 pybaseutils是个人开发常用的python库，集成了python等常用的算法
 
-- 安装方法1：pip install 
-- pybaseutils (有延时，可能不是最新版本)
-- 安装方法2：pip install --upgrade  -i https://pypi.org/simple pybaseutils (从pypi源下载最新版本)
+- 安装方法1：pip install --upgrade -i https://pypi.tuna.tsinghua.edu.cn/simple pybaseutils (有延时，可能不是最新版本)
+- 安装方法2：pip install --upgrade -i https://pypi.org/simple pybaseutils (从pypi源下载最新版本)
 
+(设置默认的镜像源：pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple)
 
 ## 一些问题修复说明
 

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils/__init__.py

@@ -8,4 +8,4 @@
 # --------------------------------------------------------
 """
 
-__version__ = '2.2.19'
+__version__ = '2.2.21'

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils/file_utils.py

@@ -43,12 +43,11 @@ def get_time(format="p"):
     :param format:
     :return:
     """
-    if format.lower() == "s":
+    if format.lower() == "s":  # 精确到秒
         # time = datetime.strftime(datetime.now(), '%Y%m%d_%H%M%S')
         time = datetime.strftime(datetime.now(), '%Y%m%d%H%M%S')
-    elif format.lower() == "p":
-        # 20200508_143059_379116
-        time = datetime.strftime(datetime.now(), '%Y%m%d_%H%M%S_%f')
+    elif format.lower() == "p":  # 精确到微妙
+        time = datetime.strftime(datetime.now(), '%Y%m%d_%H%M%S_%f')  # 20200508_143059_379116
         time = time[:-2]
     else:
         time = (str(datetime.now())[:-10]).replace(' ', '-').replace(':', '-')
@@ -108,14 +107,14 @@ def combine_flags(flags: list, use_time=True, info=True):
     return out_flags
 
 
-def write_file(file, data):
+def write_file(file, data, mode='wb'):
     """写二进制数据"""
-    with open(file, 'wb') as f: f.write(data)
+    with open(file, mode) as f: f.write(data)
 
 
-def read_file(file):
+def read_file(file, mode='rb'):
     """读取二进制数据"""
-    with open(file, 'rb') as f: key = f.read()
+    with open(file, mode) as f: key = f.read()
     return key
 
 
@@ -701,7 +700,7 @@ def create_file_path(filename):
     return out_path
 
 
-def get_sub_paths(input_dir):
+def get_sub_paths(input_dir, abspath=False):
     """
     当前路径下所有子目录
     :param input_dir:
@@ -715,6 +714,7 @@ def get_sub_paths(input_dir):
     # print(dirs)   # 当前路径下所有子目录
     # print(files)  # 当前路径下所有非目录子文件
     sub_list.sort()
+    if abspath: sub_list = [os.path.join(input_dir, p) for p in sub_list]
     return sub_list
 
 

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils/font_style/__init__.py

@@ -6,7 +6,7 @@
 """
 import os
 
-# TODO 如果字体缺失，可以把字体拷贝到系统字体目录：cp -r fonts/* ~/.local/share/fonts/
+# TODO 如果字体缺失，可以把字体拷贝到系统字体目录：cp -r font_style/*
 # windows: C:\Windows\Fonts可查看系统支持的字体
 # Linux  : /usr/share/fonts/truetype 或者~/.local/share/fonts中 可查看系统支持的字体
 

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils/font_utils.py

@@ -58,9 +58,6 @@ class FontType(object):
         return self.font_type
 
 
-# font_type = FontType()
-
-
 def get_font_type(size, font=""):
     """
     Windows字体路径      : /usr/share/fonts/楷体.ttf
@@ -101,8 +98,7 @@ def draw_image_text(image, point, text, style="楷体", size=20, color=(255, 255
         font = ImageFont.truetype(style, size=size, encoding="utf-8")
     else:
         # simhei.ttf 是字体，你如果没有字体，需要下载
-        font_type.set_font_style(style=style, size=size)
-        font = font_type.get_font_type()
+        font = get_font_type(size=size)
     draw.text(point, text, fill=color, font=font)
     image = np.asarray(image)
     return image
@@ -211,6 +207,24 @@ def get_font_char(font_file, only_chinese=False):
     return fonts
 
 
+def set_pyplot_font(font="simhei"):
+    """
+    设置 seaborn，matplotlib等默认字体
+    :param font: 字体路径或者名称，或者使用默认的,SimHei
+    :return:
+    """
+    import seaborn
+    import matplotlib.pyplot as plt
+    import matplotlib.font_manager as fm
+    font_file = font if os.path.isfile(font) else FONT_TABLES[font]
+    font = os.path.basename(font_file).split(".")[0]
+    fm.fontManager.addfont(font_file)
+    plt.rcParams['font.sans-serif'] = [font]
+    plt.rcParams['axes.unicode_minus'] = False
+    seaborn.set_theme(font=font)  # 增加一行代码, 设置 seaborn 使用的字体文件
+    return font_file
+
+
 def draw_font_example():
     size = 512
     string = "我"

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils/image_utils.py

@@ -1794,6 +1794,17 @@ def nms_boxes_cv2(boxes: np.ndarray, scores: np.ndarray, labels: np.ndarray, sco
     return index
 
 
+def image2bytes(image):
+    """将numpy数据转换为JPEG编码格式的字节数据"""
+    _, image = cv2.imencode('.jpg', image)
+    return image.tobytes()
+
+
+def bytes2image(bytes):
+    image = cv2.imdecode(np.frombuffer(bytes, np.uint8), cv2.IMREAD_COLOR)
+    return image
+
+
 def file2base64(file):
     image_base64 = base64.b64encode(open(file, 'rb').read()).decode()
     return image_base64
@@ -2851,6 +2862,29 @@ def get_image_block(image, grid=[3, 3], same=False):
     return image_block
 
 
+def apply_mosaic(image, boxes, radius=8, scale=[1.0, 1.0]):
+    """
+    马赛克
+    :param image: BGR image
+    :param boxes: 人脸框(xmin,ymin,xmax,ymax)
+    :param radius: 马赛克强度
+    :param scale: 人脸框扩大范围
+    :return:
+    """
+    h, w = image.shape[:2]
+    boxes = np.asarray(boxes, dtype=np.int32)
+    boxes = extend_xyxy(boxes, scale=scale, valid_range=(0, 0, w, h))
+    for box in boxes:
+        x1, y1, x2, y2 = box
+        roi = image[y1:y2, x1:x2]
+        size = (max((x2 - x1) // radius, 5), max((y2 - y1) // radius, 5))
+        img = cv2.resize(roi, size)
+        # img = cv2.resize(roi, (radius, radius))
+        img = cv2.resize(img, (x2 - x1, y2 - y1), interpolation=cv2.INTER_NEAREST)
+        image[y1:y2, x1:x2] = img
+    return image
+
+
 def image_composite(image: np.ndarray, alpha: np.ndarray, bg_img=(219, 142, 67)):
     """
     图像融合：合成图 = 前景*alpha+背景*(1-alpha)
@@ -3004,6 +3038,7 @@ def get_video_info(video_cap: cv2.VideoCapture, vis=True):
     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)
+    fps = math.ceil(fps)
     if vis: print("read video:width:{},height:{},fps:{},num_frames:{}".format(width, height, fps, num_frames))
     return width, height, num_frames, fps
 

pybaseutils-2.2.21/pybaseutils/web/app_flask_image.py

@@ -0,0 +1,94 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-02-27 16:03:39
+# @Brief  : 基于pywebio实现图像处理web小工具
+#           运行方法：python app_webio_image.py
+# --------------------------------------------------------
+"""
+import os
+import cv2
+import numpy as np
+from pywebio.input import file_upload, input_group, actions
+from pywebio.output import put_image, put_buttons, put_row, put_column, use_scope, clear, put_text, put_scope
+from pywebio.pin import pin, put_input
+from pywebio.session import set_env
+from pywebio.pin import *
+
+
+def blur_image(image_path):
+    # 读取图像
+    image = cv2.imread(image_path)
+    # 应用高斯模糊
+    blurred_image = cv2.GaussianBlur(image, (15, 15), 0)
+    # 保存模糊后的图像
+    blurred_image_path = f"blurred_{os.path.basename(image_path)}"
+    cv2.imwrite(blurred_image_path, blurred_image)
+    return blurred_image_path
+
+
+def show_image(image_path):
+    with use_scope('image_display', clear=True):
+        put_image(open(image_path, 'rb').read())
+
+
+def handle_single_image(file_info):
+    file_name = file_info['filename']
+    file_content = file_info['content']
+    with open(file_name, 'wb') as f:
+        f.write(file_content)
+    blurred_image_path = blur_image(file_name)
+    show_image(blurred_image_path)
+
+
+def handle_multiple_images(directory):
+    global current_image_index
+    current_image_index = 0
+    images = [os.path.join(directory, f) for f in os.listdir(directory) if
+              f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+    if not images:
+        put_text("No images found in the selected directory.")
+        return
+    current_image_path = images[current_image_index]
+    blurred_image_path = blur_image(current_image_path)
+    show_image(blurred_image_path)
+    put_buttons(['Previous', 'Next'], onclick=[lambda: switch_image(images, -1), lambda: switch_image(images, 1)])
+
+
+def switch_image(images, direction):
+    global current_image_index
+    current_image_index += direction
+    current_image_index = max(0, min(current_image_index, len(images) - 1))
+    current_image_path = images[current_image_index]
+    blurred_image_path = blur_image(current_image_path)
+    show_image(blurred_image_path)
+
+
+def main():
+    set_env(output_max_width='80%')
+    put_column([
+        put_row([
+            put_buttons(['选择图片'], onclick=lambda: pin.update(
+                name=file_upload(label='选择图片', accept='image/*', help_text='请选择一张图片'))),
+            put_buttons(['选择目录'], onclick=lambda: pin.update(name=input_group("选择目录", [
+                input('directory', type='text', name='directory', placeholder='请输入目录路径')]))),
+        ]),
+        put_scope('image_display')
+    ])
+
+    while True:
+        event = pin_wait_change('name')
+        if event['name'] == 'value':
+            value = event['value']
+            if isinstance(value, dict):  # 单张图片
+                handle_single_image(value)
+            elif isinstance(value, str):  # 目录
+                handle_multiple_images(value)
+
+
+if __name__ == '__main__':
+    from pywebio.platform.flask import start_server
+
+    start_server(main, port=8080, debug=True)

pybaseutils-2.2.19/pybaseutils/web/st_image_web.py → pybaseutils-2.2.21/pybaseutils/web/app_stweb_image.py

@@ -4,7 +4,8 @@
 # @Author : Pan
 # @E-mail :
 # @Date   : 2025-02-18 16:10:49
-# @Brief  :
+# @Brief  : 基于streamlit实现图像处理web小工具
+#           运行方法：streamlit run app_stweb_image.py
 # --------------------------------------------------------
 """
 

pybaseutils-2.2.21/pybaseutils/web/app_webio_image.py

@@ -0,0 +1,94 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-02-27 16:03:39
+# @Brief  : 基于pywebio实现图像处理web小工具
+#           运行方法：python app_webio_image.py
+# --------------------------------------------------------
+"""
+import os
+import cv2
+import numpy as np
+from pywebio.input import file_upload, input_group, actions
+from pywebio.output import put_image, put_buttons, put_row, put_column, use_scope, clear, put_text, put_scope
+from pywebio.pin import pin, put_input
+from pywebio.session import set_env
+from pywebio.pin import *
+
+
+def blur_image(image_path):
+    # 读取图像
+    image = cv2.imread(image_path)
+    # 应用高斯模糊
+    blurred_image = cv2.GaussianBlur(image, (15, 15), 0)
+    # 保存模糊后的图像
+    blurred_image_path = f"blurred_{os.path.basename(image_path)}"
+    cv2.imwrite(blurred_image_path, blurred_image)
+    return blurred_image_path
+
+
+def show_image(image_path):
+    with use_scope('image_display', clear=True):
+        put_image(open(image_path, 'rb').read())
+
+
+def handle_single_image(file_info):
+    file_name = file_info['filename']
+    file_content = file_info['content']
+    with open(file_name, 'wb') as f:
+        f.write(file_content)
+    blurred_image_path = blur_image(file_name)
+    show_image(blurred_image_path)
+
+
+def handle_multiple_images(directory):
+    global current_image_index
+    current_image_index = 0
+    images = [os.path.join(directory, f) for f in os.listdir(directory) if
+              f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+    if not images:
+        put_text("No images found in the selected directory.")
+        return
+    current_image_path = images[current_image_index]
+    blurred_image_path = blur_image(current_image_path)
+    show_image(blurred_image_path)
+    put_buttons(['Previous', 'Next'], onclick=[lambda: switch_image(images, -1), lambda: switch_image(images, 1)])
+
+
+def switch_image(images, direction):
+    global current_image_index
+    current_image_index += direction
+    current_image_index = max(0, min(current_image_index, len(images) - 1))
+    current_image_path = images[current_image_index]
+    blurred_image_path = blur_image(current_image_path)
+    show_image(blurred_image_path)
+
+
+def main():
+    set_env(output_max_width='80%')
+    put_column([
+        put_row([
+            put_buttons(['选择图片'], onclick=lambda: pin.update(
+                name=file_upload(label='选择图片', accept='image/*', help_text='请选择一张图片'))),
+            put_buttons(['选择目录'], onclick=lambda: pin.update(name=input_group("选择目录", [
+                input('directory', type='text', name='directory', placeholder='请输入目录路径')]))),
+        ]),
+        put_scope('image_display')
+    ])
+
+    while True:
+        event = pin_wait_change('name')
+        if event['name'] == 'value':
+            value = event['value']
+            if isinstance(value, dict):  # 单张图片
+                handle_single_image(value)
+            elif isinstance(value, str):  # 目录
+                handle_multiple_images(value)
+
+
+if __name__ == '__main__':
+    from pywebio.platform.flask import start_server
+
+    start_server(main, port=8080, debug=True)

pybaseutils-2.2.21/pybaseutils/web/demo.py

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-02-27 16:03:39
+# @Brief  : 基于pywebio实现图像处理web小工具
+#           运行方法：python app_webio_image.py
+# --------------------------------------------------------
+"""
+
+if __name__ == '__main__':
+    pass

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pybaseutils
-Version: 2.2.19
+Version: 2.2.21
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
@@ -36,11 +36,15 @@ pybaseutils(Python)
 
 pybaseutils是个人开发常用的python库，集成了python等常用的算法
 
--  安装方法1：pip install
--  pybaseutils (有延时，可能不是最新版本)
+-  安装方法1：pip install –upgrade -i
+   https://pypi.tuna.tsinghua.edu.cn/simple pybaseutils
+   (有延时，可能不是最新版本)
 -  安装方法2：pip install –upgrade -i https://pypi.org/simple
    pybaseutils (从pypi源下载最新版本)
 
+(设置默认的镜像源：pip config set global.index-url
+https://pypi.tuna.tsinghua.edu.cn/simple)
+
 一些问题修复说明
 ----------------
 

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/pybaseutils.egg-info/SOURCES.txt

@@ -119,8 +119,12 @@ pybaseutils/transforms/augment_utils.py
 pybaseutils/transforms/face_alignment.py
 pybaseutils/transforms/transform_utils.py
 pybaseutils/web/__init__.py
-pybaseutils/web/st_image_web.py
+pybaseutils/web/app_flask_image.py
+pybaseutils/web/app_stweb_image.py
+pybaseutils/web/app_webio_image.py
+pybaseutils/web/demo.py
 test_py/__init__.py
+test_py/audio_demo1.py
 test_py/class_attribute.py
 test_py/class_names.py
 test_py/date_dataset.py
@@ -245,8 +249,9 @@ test_py/edit_distance/__init__.py
 test_py/edit_distance/demo.py
 test_py/edit_distance/text_matching.py
 test_py/flask_demo/__init__.py
-test_py/flask_demo/func.py
-test_py/flask_demo/server.py
+test_py/flask_demo/app.py
+test_py/flask_demo/utils/__init__.py
+test_py/flask_demo/utils/utils.py
 test_py/image_correction/__init__.py
 test_py/image_correction/demo_correction_v1.py
 test_py/image_correction/demo_correction_v2.py

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/test_py/aije/demo_labelme_crop.py

@@ -12,7 +12,7 @@ from pybaseutils.dataloader import parser_labelme
 from pybaseutils import image_utils, file_utils
 
 
-def save_object_crops(data_info, out_dir, class_name=None, scale=[], square=False,
+def save_object_crops(data_info, out_dir, class_name=None, target_name=None, scale=[], square=False,
                       padding=False, min_size=20 * 20 * 3, flag="", vis=False):
     """
     对VOC的数据目标进行裁剪
@@ -51,22 +51,16 @@ def save_object_crops(data_info, out_dir, class_name=None, scale=[], square=Fals
         name = class_name[int(labels[i])] if class_name else labels[i]
         if out_dir:
             file_name = "{}_{:0=4d}_{}.jpg".format(image_id, i, flag) if flag else "{}_{:0=4d}.jpg".format(image_id, i)
-            img_file = file_utils.create_dir(out_dir, name, file_name)
+            if target_name:
+                out_dir_ = out_dir if name in target_name else os.path.join(out_dir, "其他")
+                img_file = file_utils.create_dir(out_dir_, name, file_name)
+            else:
+                img_file = file_utils.create_dir(out_dir, name, file_name)
             cv2.imwrite(img_file, img)
         if vis: image_utils.cv_show_image("crop", img, use_rgb=False, delay=0)
 
 
-if __name__ == "__main__":
-    """
-    对VOC的数据目标进行裁剪
-    室内：['主杆', '从此进出标示牌', '其他鞋', '吊物绳', '在此工作标示牌', '垫子', '安全带', '安全帽', '安全绳',
-          '导线头', '尖嘴钳', '工具袋', '手', '扳手', '抹布', '未穿工作服','柱式绝缘子', '止步高压危险标示牌',
-          '绝缘手套', '绝缘鞋', '脚扣', '螺丝', '身穿工作服', '遮拦杆', '铁架', '铝扎线']
-    室外: []
-    """
-    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-v2-det/dataset-v31/images"
-    class_name = None  # 室内\
-    out_dir = os.path.join(os.path.dirname(anno_dir), "crops")
+def crop_dataset(anno_dir, out_dir=None, class_name=None, target_name=None):
     dataset = parser_labelme.LabelMeDatasets(filename=None,
                                              data_root=None,
                                              anno_dir=anno_dir,
@@ -83,4 +77,46 @@ if __name__ == "__main__":
     # scale = None
     for i in tqdm(range(len(dataset))):
         data_info = dataset.__getitem__(i)
-        save_object_crops(data_info, out_dir, class_name=class_name, scale=scale, flag=flag, vis=False)
+        save_object_crops(data_info, out_dir, class_name=class_name, target_name=target_name, scale=scale, flag=flag,
+                          vis=False)
+
+
+def get_sub_dataset(data_root):
+    sub_paths = file_utils.get_sub_paths(data_root, abspath=True)
+    datasets = []
+    for sub_path in sub_paths:
+        data_list = file_utils.get_sub_paths(sub_path, abspath=True)
+        data_list = [os.path.join(p, "images") for p in data_list if os.path.exists(os.path.join(p, "images"))]
+        datasets += data_list
+    return datasets
+
+
+if __name__ == "__main__":
+    """
+    对labelme的数据目标进行裁剪，用于制作分类数据集
+    """
+
+    target_name = ["身穿工作服", "未穿工作服",
+                   "绝缘鞋", "脚穿绝缘鞋", "长筒靴", "脚穿长筒靴", "其他鞋", "脚穿其他鞋",
+                   "手", "绝缘手套", "手穿绝缘手套", "棉纱手套", "手穿棉纱手套", "其他手套", "手穿其他手套",
+                   "拨动开关分闸", "拨动开关合闸",
+                   "电容柜开关合闸", "电容柜开关分闸",
+                   "低压侧总刀闸合闸", "低压侧总刀闸分闸",
+                   "面板单键开关断开", "面板单键开关合上",
+                   "面板双键开关断开", "面板双键开关合上",
+                   "表箱关", "表箱开", "手", "其他",
+                   ]
+    datasets = [
+        "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/test-v2/dataset-v01/images",
+        "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/test-v2/dataset-v02/images",
+    ]
+    # dataroot = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-v2-det"
+    # datasets = get_sub_dataset(dataroot)
+    class_name = []
+    for anno_dir in datasets:
+        print(anno_dir)
+        out_dir = os.path.join(os.path.dirname(anno_dir), "crops")
+        if os.path.exists(out_dir) and out_dir.endswith("crops"): file_utils.remove_dir(out_dir)
+        file_utils.create_dir(out_dir)
+        file = file_utils.write_file(os.path.join(out_dir, "分类数据集，请勿删除.txt"), data="", mode="w")
+        crop_dataset(anno_dir, out_dir=out_dir, class_name=class_name, target_name=target_name)

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/test_py/aije/demo_video_aije.py

@@ -8,7 +8,6 @@
 import os
 import cv2
 import numpy as np
-from xpinyin import Pinyin
 from tqdm import tqdm
 from pybaseutils import file_utils, image_utils
 from pybaseutils.cvutils import video_utils, monitor
@@ -73,7 +72,8 @@ def video2frames_similarity(video_file, out_dir=None, func=None, interval=1, thr
 
 # 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}
-thresh_dict = {"第一视角": 0.50, "主视角": 0.50, "平视": 0.30, "俯视": 0.30, "右视": 0.35, "左视": 0.35, "全景": 0.30}
+thresh_dict = {"第一视角": 0.50, "主视角": 0.50, "主视": 0.50, "平视": 0.30, "俯视": 0.30, "右视": 0.35, "左视": 0.35,
+               "全景": 0.30}
 
 
 # thresh_dict = {"1号位": 0.58, "平视": 0.35, "全景": 0.35, "俯视": 0.35, "右视": 0.25, "左视": 0.3}
@@ -92,7 +92,7 @@ def video2frames_demo(root, out, prefix="", thresh=0.2):
 
 
 if __name__ == "__main__":
-    root = "/home/PKing/nasdata/dataset-dmai/AIJE/技能人才系统_数据集管理/01-江门四维数据/01-37道考题视频/江门四维2024-12-06-检查帽撑/video"
+    root = "/home/PKing/Downloads/video/08-更换熔丝"
     prefix = ""
     out = root + "-train"
     video2frames_demo(root, out, prefix=prefix)

pybaseutils-2.2.21/test_py/audio_demo1.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+"""
+    @Author : PKing
+    @E-mail :
+    @Date   : 2024-02-05 18:19:18
+    @Brief  :
+"""
+import cv2
+import numpy as np
+from pybaseutils import file_utils, image_utils, plot_utils, numpy_utils
+from pybaseutils.audio import audio_utils
+
+import pyaudio
+import wave
+import time
+
+file = "/home/PKing/Videos/dde-introduction.wav"
+# 打开WAV文档
+wf = wave.open(file, "rb")
+
+# 实例化一个PyAudio对象
+pa = pyaudio.PyAudio()
+# 打开声卡
+stream = pa.open(format=pa.get_format_from_width(wf.getsampwidth()),  # 从wf中获取采样深度
+                 channels=wf.getnchannels(),  # 从wf中获取声道数
+                 rate=wf.getframerate(),  # 从wf中获取采样率
+                 output=True)  # 设置为输出
+count = 0
+while count < 8 * 500:
+    time.sleep(0.001)
+    data = wf.readframes(15)
+    stream.write(data)

pybaseutils-2.2.21/test_py/demo1.py

@@ -0,0 +1,14 @@
+import requests
+from urllib.parse import urljoin, urlparse
+from bs4 import BeautifulSoup
+from pybaseutils import file_utils, json_utils
+import glob
+
+# 使用示例
+if __name__ == "__main__":
+    data_root = "http://aije-mvp-nginx.partner.dm-ai.com/req-resp/aije-job-m8ch701s-awbg/*.json"
+    # data_root = "/media/PKing/新加卷/SDK/base-utils/data/person/*.json"
+    file = glob.glob(data_root)
+    print(file)
+    data =json_utils.read_json_data("http://aije-mvp-nginx.partner.dm-ai.com/req-resp/aije-job-m8ch701s-awbg/nlp-12.json")
+    print(data)

{pybaseutils-2.2.19 → pybaseutils-2.2.21}/test_py/demo_lableme_vis.py

@@ -28,9 +28,9 @@ if __name__ == "__main__":
     anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-v2-det/dataset-v25/json"
     anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-negetive/dataset-v02/json"
     # anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-date/date-det/dataset-v01/images"
-    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-v2-det/09-更换熔断器/dataset-v20/images"
+    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/train-v2-其他/check-helmet/images"
     names = None
-    # names = {'unique': 0}
+    # names = {'手拿安全帽帽撑': 0}
     dataset = parser_labelme.LabelMeDatasets(filename=None,
                                              data_root=None,
                                              anno_dir=anno_dir,
@@ -51,6 +51,6 @@ if __name__ == "__main__":
         if class_name: labels = [class_name[l] for l in labels]
         image_file = data["image_file"]
         anno_file = os.path.join("masker", "{}.json".format(os.path.basename(image_file).split(".")[0]))
-        print(image_file, labels, points)
+        print(image_file, labels)
         result = parser_labelme.show_target_image(image, bboxes, labels, points, thickness=2)
         # image_utils.save_image("./"+os.path.basename(image_file), result)