pybaseutils 2.4.5__tar.gz → 2.4.7__tar.gz

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/PKG-INFO

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

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/__init__.py

@@ -8,4 +8,4 @@
 # --------------------------------------------------------
 """
 
-__version__ = '2.4.5'
+__version__ = '2.4.7'

pybaseutils-2.4.7/pybaseutils/base/producer_consumer.py

@@ -0,0 +1,138 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-10-29 16:52:22
+# @Brief  :
+# --------------------------------------------------------
+"""
+import os
+import numpy as np
+import time
+import threading
+from typing import List, Dict, Callable
+from pybaseutils import thread_utils, file_utils, image_utils, base64_utils
+from pybaseutils.cvutils import video_utils
+from pybaseutils.base import list_queue
+
+thread_lock = threading.Lock()
+
+
+class ProducerConsumer:
+    def __init__(self, winsize=3, overlap=0.0, max_workers=2, **kwargs) -> None:
+        """
+        :param winsize:
+        :param overlap:
+        :param max_workers:
+        :param kwargs:
+        """
+        self.tag = self.__class__.__name__
+        self.winsize = winsize
+        self.overlap = overlap
+        self.max_workers = max_workers
+        self.kwargs = kwargs
+        self.producer = list_queue.Queue(maxsize=self.winsize)  # 生产者
+        self.consumer = list_queue.Queue(maxsize=self.winsize)  # 消费者
+        self.producer_end = False  # 生产者是否结束
+        self.consumer_end = False  # 消费者是否结束
+
+    def stop_producer(self):
+        """
+        停止生产者线程
+        :return:
+        """
+        self.producer_end = True
+
+    def stop_consumer(self):
+        """
+        停止消费者线程
+        :return:
+        """
+        self.consumer_end = True
+
+    def task(self, data: List, *args, **kwargs):
+        """
+        任务函数，处理输入数据
+        :param data: 输入数据
+        :return: 处理结果
+        """
+        print(f"args  ={args},kwargs={kwargs}")
+        time.sleep(5)
+        result = "收到消息：" + file_utils.get_time(format="s")
+        return result
+
+    def task_consumer(self, task, *args, **kwargs):
+        """消费者任务"""
+        print(f"start consumer thread")
+        self.consumer_end = False
+        pool = thread_utils.ThreadPool(max_workers=self.max_workers)
+        future = []
+        while True:
+            winsize = self.producer.qsize() if self.producer_end else self.winsize
+            while self.producer.qsize() >= winsize and len(future) < self.max_workers:
+                data = self.producer.get_window(winsize=winsize, overlap=self.overlap, block=False)  # 提取窗口数据
+                f = pool.submit(task, data, args, **kwargs)
+                future.append(f)
+                print(f"producer={self.producer.qsize()} consumer={self.consumer.qsize()} "
+                      f"winsize={winsize} future={len(future)}")
+            if not future: time.sleep(0.1)
+            tmp = []
+            for f in future:
+                try:
+                    self.consumer.put(f.result(timeout=0.1))
+                except Exception as e:
+                    tmp.append(f)
+            future = tmp
+            # 生产者队列空，且停止发送数据(pend=True)
+            if len(future) == 0 and self.producer_end and self.producer.qsize() == 0:
+                print(f"finish consumer thread")
+                break
+        self.consumer_end = True  # 消费者停止处理数据
+
+    def start_consumer(self, task: Callable, *args, **kwargs):
+        """启动消费者线程
+        :param task: 任务函数
+        :param args: 任务函数参数
+        :param kwargs: 任务函数参数
+        :return:
+        """
+        thread = threading.Thread(target=self.task_consumer, args=args, kwargs={"task": task, **kwargs})
+        thread.daemon = True
+        thread.start()
+
+    def example(self, video, realtime=False, freq=2, vis=True):
+        """
+        :param video: 视频文件路径或者摄像头ID
+        :param realtime: 是否实时处理视频帧，实时处理会丢弃很多视频帧，非实时会处理所有视频帧，但视频播放会很卡顿
+        :param freq:
+        :return:
+        """
+        # TODO 打开视频文件或摄像头
+        w, h, num_frames, fps = image_utils.get_video_info(video)
+        interval = int(fps / freq) if fps > 0 else 1
+        video_cap = video_utils.video_iterator(video, save_video=None)
+        self.start_consumer(task=self.task, param1="123", param2="456")  # 启动消费者线程
+        # TODO 主线程处理视频帧
+        for data_info in video_cap:
+            image = data_info["frame"][:, :, ::-1]  # BGR to RGB
+            image = image_utils.resize_image(image, size=(None, 640))
+            data_info["frame"] = image
+            # TODO 视频抽帧，放入生产者队列，非实时处理会阻塞，实时处理会丢弃旧的视频帧
+            if data_info['count'] % interval == 0:
+                # data_info['text'] = text
+                # data_info['freq'] = freq
+                self.producer.put(data_info, block=not realtime)
+            if vis: image_utils.show_image("image", image, delay=20)
+            if data_info["finish"]: self.stop_producer()  # 标记生产者是否结束
+            while self.consumer.qsize() > 0 or (self.producer_end and not self.consumer_end):
+                result = self.consumer.pop(block=False, timeout=0.005)
+                if result: print(result)
+
+
+if __name__ == '__main__':
+    # video_file = "/home/PKing/nasdata/Project/LLM/MLLM-Factory/data/video2.mp4"
+    video_file = "/home/PKing/Videos/aije-data/检查绝缘手套-nodate-35s.mp4"
+    freq = 2
+    infer = ProducerConsumer(winsize=freq * 3, max_workers=1)
+    result = infer.example(video=video_file, freq=2)

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/cvutils/video_utils.py

@@ -61,12 +61,12 @@ def video2gif(video_file, gif_file=None, func=None, interval=1, fps=-1, use_pil=
 
 def get_video_sampling(freq, time, fps, random=False):
     """
-    获得抽帧时间点和索引
+    根据抽帧频率和时间范围，获得抽帧时间点和索引
     :param freq: 抽帧频率
     :param time: time: 开始播放时间time[0]，结束播放时间time[1]，单位秒S
     :param fps: 视频帧率FPS
     :param random: 是否随机抽帧(在每一秒内随机抽freq帧)
-    :return:
+    :return: times,index
     """
     if freq <= 0: freq = fps
     if random:
@@ -79,7 +79,8 @@ def get_video_sampling(freq, time, fps, random=False):
         times = np.concatenate(times)
         # times = np.sort(np.random.uniform(time[0], time[1], size=freq))
     else:
-        times = np.arange(time[0], time[1], 1 / freq)
+        times = np.arange(time[0], time[1], 1 / freq)  # (
+        # 开始时间, 结束时间, 抽帧时间间隔)
     index = times * fps
     index = index.astype(int)
     return times, index
@@ -338,7 +339,7 @@ def video_capture(video_file: int or str, save_video: str or int = None, interva
         video_writer.release()
 
 
-def video_iterator(video_file: int | str, save_video: str or int = None, interval=1, size=(), freq=0,
+def video_iterator(video_file, save_video: str or int = None, interval=1, size=(), freq=0,
                    task: Callable = None, vis=False, **kwargs):
     """
     读取摄像头或者视频流迭代器
@@ -450,3 +451,7 @@ if __name__ == "__main__":
     # video2video(video_file, dst_file, vis=True)
     video_file = "/home/PKing/Videos/aije-data/检查绝缘棒.mp4"
     load_video(video_file, time=(0, -1), freq=2, size=(), use_rgb=False)
+
+    times, index = get_video_sampling(freq=4, time=(0, 4), fps=20, random=False)
+    print(times)
+    print(index)

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/dataloader/parser_video_folder.py

@@ -99,7 +99,6 @@ class VideoFolderDataset(parser_image_folder.FolderDataset):
         :param video_file:
         :param size:
         :param use_rgb:
-        :param duration: 最大时长
         :param freq: 抽帧频率
         :param use_cut:
         :return:
@@ -108,7 +107,10 @@ class VideoFolderDataset(parser_image_folder.FolderDataset):
         width, height, numFrames, fps = video_utils.get_video_info(video_cap, disp=False)
         time = (0, numFrames / fps)
         video_time, video_idx = video_utils.get_video_sampling(self.freq, time, fps, random=shuffle)
-        # assert len(video_idx) == max_nums, "video_idx len error:{}".format(video_idx)
+        # TODO 居中裁剪
+        clip = (int(len(video_idx) / 2 - self.seq_len / 2), int(len(video_idx) / 2 + self.seq_len / 2))
+        clip = (max(clip[0], 0), min(clip[1], len(video_idx)))
+        video_idx = video_idx[clip[0]:clip[1]]
         frames = []
         for count in video_idx:
             video_cap.set(cv2.CAP_PROP_POS_FRAMES, count)
@@ -120,9 +122,10 @@ class VideoFolderDataset(parser_image_folder.FolderDataset):
         if len(frames) < 3: return []  # 帧太少，无效视频
         if len(frames) < self.seq_len:
             pad = self.seq_len - len(frames)
-            images = [frames[-1].copy()] * pad
+            images = [frames[-1]] * pad
             frames += images
-        frames = frames[:self.seq_len]
+        frames = frames[0:self.seq_len]
+        assert len(frames) == self.seq_len, "frames size error:{}".format(len(frames))
         return frames
 
 

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/dataloader/parser_video_text.py

@@ -94,7 +94,6 @@ class VideoTextDataset(parser_image_text.TextDataset):
         :param video_file:
         :param size:
         :param use_rgb:
-        :param duration: 最大时长
         :param freq: 抽帧频率
         :param use_cut:
         :return:
@@ -103,7 +102,10 @@ class VideoTextDataset(parser_image_text.TextDataset):
         width, height, numFrames, fps = video_utils.get_video_info(video_cap, disp=False)
         time = (0, numFrames / fps)
         video_time, video_idx = video_utils.get_video_sampling(self.freq, time, fps, random=shuffle)
-        # assert len(video_idx) == max_nums, "video_idx len error:{}".format(video_idx)
+        # TODO 居中裁剪
+        clip = (int(len(video_idx) / 2 - self.seq_len / 2), int(len(video_idx) / 2 + self.seq_len / 2))
+        clip = (max(clip[0], 0), min(clip[1], len(video_idx)))
+        video_idx = video_idx[clip[0]:clip[1]]
         frames = []
         for count in video_idx:
             video_cap.set(cv2.CAP_PROP_POS_FRAMES, count)
@@ -115,9 +117,10 @@ class VideoTextDataset(parser_image_text.TextDataset):
         if len(frames) < 3: return []  # 帧太少，无效视频
         if len(frames) < self.seq_len:
             pad = self.seq_len - len(frames)
-            images = [frames[-1].copy()] * pad
+            images = [frames[-1]] * pad
             frames += images
-        frames = frames[:self.seq_len]
+        frames = frames[0:self.seq_len]
+        assert len(frames) == self.seq_len, "frames size error:{}".format(len(frames))
         return frames
 
 
@@ -126,10 +129,11 @@ if __name__ == '__main__':
     from classifier.dataset import build_dataset
     from pybaseutils import image_utils
 
-    data_file = ["/home/PKing/nasdata/tmp/UCF101/UCF101/file_test.txt"]
+    data_file = ["/home/PKing/nasdata/tmp/UCF101/UCF101/val.txt"]
     batch_size = 1
     input_size = [224, 224]
-    trans_type = "train_video"
+    # trans_type = "train_video"
+    trans_type = "test_video"
     transform = build_transform.image_transform(input_size=input_size, trans_type=trans_type)
     cfg = {"duration": 5, "freq": 2}
     class_name = ["ApplyEyeMakeup", "ApplyLipstick", "Archery"]
@@ -143,7 +147,7 @@ if __name__ == '__main__':
                                cfg=cfg,
                                disp=True)
     for i in range(len(dataset)):
-        data = dataset.__getitem__(10)
+        data = dataset.__getitem__(0)
         video_file, image, label = data["file"], data["image"], data["label"]
         image = image.transpose((1, 2, 3, 0))
         image = np.asarray((image * 0.5 + 0.5) * 255, dtype=np.uint8)

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/image_utils.py

@@ -719,26 +719,40 @@ def resize_image_clip(image, clip_max=1920, interpolation=cv2.INTER_LINEAR):
     return image
 
 
+def autosize(size, dsize):
+    """
+    自动调整图像大小
+    :param size : 原始图像大小 (width, height) image.shape[:2][::-1] or image.size
+    :param dsize: 目标图像大小 (width, height)
+    :return: 调整后的图像大小 (width, height)
+    """
+    if not dsize: return size
+    dsize = (dsize, dsize) if isinstance(dsize, numbers.Number) else dsize
+    dw, dh = dsize
+    sw, sh = size
+    if (dh is None) and (dw is None):  # 错误写法：resize_height and resize_width is None
+        return None
+    if dh is None:
+        dh = int(sh * dw / sw)
+    elif dw is None:
+        dw = int(sw * dh / sh)
+    dsize = (int(dw), int(dh))
+    return dsize
+
+
 def resize_image(image, size: Tuple, interpolation=cv2.INTER_LINEAR):
     """
     tf.image.resize_images(images,size),images=[batch, height, width, channels],size=(new_height, new_width)
     cv2.resize(image, dsize=(width, height)),与image.shape相反
     images[50,10]与image.shape的原理相同，它表示的是image=(y=50,x=10)
     :param image:
-    :param size: (width,height)
+    :param size: (width,height) image.shape[:2][::-1] or image.size
     :return:
     """
     if not size: return image
-    size = (size, size) if len(size) == 1 else size
-    dw, dh = size
-    sh, sw = image.shape[:2]
-    if (dh is None) and (dw is None):  # 错误写法：resize_height and resize_width is None
-        return image
-    if dh is None:
-        dh = int(sh * dw / sw)
-    elif dw is None:
-        dw = int(sw * dh / sh)
-    image = cv2.resize(image, dsize=(int(dw), int(dh)), interpolation=interpolation)
+    ssize = image.shape[:2][::-1]
+    dsize = autosize(ssize, size)
+    if size and dsize: image = cv2.resize(image, dsize=dsize, interpolation=interpolation)
     return image
 
 
@@ -2410,6 +2424,23 @@ def boxes2polygons(boxes: np.ndarray or List[np.ndarray]):
     return polygons
 
 
+def get_point_circle_contour(points, r, nums=16):
+    """
+    以给定的中心坐标和半径生成圆的轮廓点
+    :param points: (n_points, 2),圆的中心坐标 (x, y)
+    :param r: 圆的半径
+    :param nums: 轮廓点的数量
+    :return: 圆的轮廓点坐标数组 (n_points,nums, 2)
+    """
+    if len(points) == 0: return points
+    if not isinstance(points, np.ndarray): points = np.asarray(points)
+    theta = np.linspace(0, 2 * np.pi, nums, endpoint=False)
+    x_contour = points[:, 0:1] + r * np.cos(theta)
+    y_contour = points[:, 1:2] + r * np.sin(theta)
+    output = np.concatenate((x_contour[:, :, None], y_contour[:, :, None]), axis=2)
+    return output
+
+
 def get_mask_iou(mask1, mask2, binarize=True, iom=False):
     """
     计算IOU=交集(A,B)/并集(A,B)
@@ -2659,7 +2690,7 @@ def draw_image_contours(image, contours: List[np.ndarray], texts=[], color=(), a
     :param contours: List[np.ndarray],每个列表是一个轮廓(num_points,1,2)
     :param texts:轮廓文本
     :param color:绘制轮廓的颜色
-    :param alpha:绘制颜色的透明度
+    :param alpha:绘制颜色的透明度 0: 轮廓填充完全透明，1: 轮廓填充完全不透明
     :param thickness:轮廓线宽
     :return:
     """
@@ -3131,7 +3162,7 @@ def image_composite(image: np.ndarray, alpha: np.ndarray, bg_img=(219, 142, 67))
     https://blog.csdn.net/guduruyu/article/details/71439733
     更有效的C++实现: https://www.aiuai.cn/aifarm1237.html
     :param image: RGB图像(uint8)
-    :param alpha: 单通道的alpha图像(uint8)
+    :param alpha: 单通道的alpha图像(uint8), 0: 轮廓填充完全透明，1: 轮廓填充完全不透明
     :param bg_img: 背景图像,可以是任意的分辨率图像，也可以指定指定纯色的背景
     :return: 返回与背景合成的图像
     """
@@ -3268,10 +3299,10 @@ def get_video_capture(video, width=None, height=None, fps=None):
     return video_cap
 
 
-def get_video_info(video_cap: int | str | cv2.VideoCapture, disp=True, **kwargs):
+def get_video_info(video_cap, disp=True, **kwargs):
     """
     获得视频的基础信息
-    :param video_cap:视频对象 或者视频文件路径
+    :param video_cap:视频对象 或者视频文件路径 int | str | cv2.VideoCapture
     :param disp: 是否显示视频信息
     :return:
     """

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/metrics/class_report.py

@@ -64,12 +64,13 @@ def get_confusion_matrix(true_labels, pred_labels, target_names=None, filename=N
     return conf_matrix
 
 
-def get_classification_report(true_labels, pred_labels, labels=None, target_names=None, output_dict=False):
+def get_metrics_report(true_labels, pred_labels, target_names=None, labels=None,
+                       output_dict=False, matrix=False, plot=False):
     """
     true_labels = [0, 1, 2, 3, 4, 1]  # Y
     pred_labels = [0, 1, 1, 2, 2, 1]  # X
     target_names = ["A", "B", "C", "D", "E"]
-    out_result = get_classification_report(true_labels, pred_labels, target_names=target_names, output_dict=False)
+    out_result = get_metrics_report(true_labels, pred_labels, target_names=target_names, output_dict=False)
     宏平均(macro avg)和微平均(micro avg)
     如果每个class的样本数量差不多,那么宏平均和微平均没有太大差异
     如果每个class的样本数量差异很大,而且你想:
@@ -77,21 +78,24 @@ def get_classification_report(true_labels, pred_labels, labels=None, target_name
     更注重样本量少的class:使用宏平均,若宏平均比微平均小,应检查样本量少的class
     :param true_labels:
     :param pred_labels:
-    :param labels:
     :param target_names:
+    :param labels:
     :param output_dict:
+    :param matrix: 是否绘制混淆矩阵
     :return:
     """
-    true_labels = np.array(true_labels, dtype=np.int32)
-    pred_labels = np.array(pred_labels, dtype=np.int32)
-    if target_names:
-        labels = list(range(len(target_names)))
+    # if target_names is None:
+    #     target_names = list(set(pred_labels) | set(true_labels))
+    # else:
+    #     true_labels = [target_names[int(i)] for i in true_labels]
+    #     pred_labels = [target_names[int(i)] for i in pred_labels]
     result = metrics.classification_report(true_labels,
                                            pred_labels,
                                            labels=labels,
                                            digits=4,
                                            target_names=target_names,
-                                           output_dict=output_dict)
+                                           output_dict=output_dict,
+                                           zero_division=0)
     if output_dict:
         macro_avg = result["macro avg"]
         accuracy = result["accuracy"]
@@ -99,12 +103,18 @@ def get_classification_report(true_labels, pred_labels, labels=None, target_name
         out_result = {"macro_avg": macro_avg, "accuracy": accuracy, "weighted_avg": weighted_avg}
         # pdf=pd.DataFrame.from_dict(result)
         # save_csv("classification_report.csv", pdf)
-
     else:
         out_result = result
+    if matrix:
+        conf_matrix = get_confusion_matrix(true_labels, pred_labels, target_names=target_names,
+                                           normalization=True, plot=plot, title="Confusion Matrix")
+        # out_result["confusion_matrix"] = conf_matrix
     return out_result
 
 
+get_classification_report = get_metrics_report
+
+
 def create_dir(parent_dir, dir1=None, filename=None):
     """
     create directory
@@ -136,13 +146,13 @@ def create_file_path(filename):
 
 
 if __name__ == "__main__":
-    true_labels = [0, 1, 2, 3, 3, 1, 1]  # Y
-    pred_labels = [1, 1, 2, 2, 2, 1, 0]  # X
-    # true_labels = [0, 1, 1, 2, 2]
-    # pred_labels = [0, 1, 1, 2, 2]
-    target_names = ["A0", "B1", "C2", "D3"]
-    confuse_file = "confuse.csv"
-    out_result = get_classification_report(true_labels, pred_labels, target_names=target_names, output_dict=False)
+    # true_labels = [0, 1, 2, 3, 3, 1, 1]  # Y
+    # pred_labels = [1, 1, 2, 2, 2, 1, 0]  # X
+    true_labels = ["A", "B", "A", "B"]
+    pred_labels = ["A", "B", "A", "B"]
+    # target_names = ["A0", "B1", "C2", "D3"]
+    target_names = None
+    confuse_file = "./confuse.csv"
+    out_result = get_classification_report(true_labels, pred_labels, target_names=target_names, output_dict=False,
+                                           matrix=True, plot=True)
     print(out_result)
-    get_confusion_matrix(true_labels, pred_labels, target_names=target_names,
-                         normalization=False, filename=confuse_file, plot=True, title="Confusion Matrix")

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils/thread_utils.py

@@ -151,7 +151,10 @@ class ThreadPool(object):
         """
          递交线程任务
          pool=thread_utils.ThreadPool(max_workers=1)
-         pool.submit(func, args=(参数1,参数2,参数3,...))
+         future = pool.submit(func, args=(参数1,参数2,参数3,...))
+         print(future.done())       # 是否完成
+         result = future.result()   # 等待线程任务完成，获取结果
+         print(future.done())       # 是否完成
         :param func:
         :param args:
         :param kwargs:
@@ -250,13 +253,14 @@ def performanceThreadPool():
     pool = ThreadPool(max_workers=2)
     future = pool.submit(consumer, "1.jpg", "2.jpg")
     future = pool.submit(consumer, "1.jpg", "2.jpg")
-    future = pool.submit(consumer, "1.jpg", "2.jpg")
+    # future = pool.submit(consumer, "1.jpg", "2.jpg")
     print(future.done())
-    result = future.result()
+    result = future.result() # 等待线程任务完成，获取结果
     print(future.done())
     print("result:{}".format(result))
     time.sleep(1)
 
+
 def performanceThreadPool2():
     from pybaseutils import time_utils
     pool = ThreadPool(max_workers=2)

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils.egg-info/PKG-INFO

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

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/pybaseutils.egg-info/SOURCES.txt

@@ -43,6 +43,7 @@ pybaseutils/audio/vad_utils.py
 pybaseutils/base/__init__.py
 pybaseutils/base/dict_queue.py
 pybaseutils/base/list_queue.py
+pybaseutils/base/producer_consumer.py
 pybaseutils/build_utils/__init__.py
 pybaseutils/build_utils/cython_utils.py
 pybaseutils/build_utils/pyarmor_utils.py
@@ -293,6 +294,8 @@ test_py/milvus_demo/hello_milvus.py
 test_py/milvus_demo/milvus_client.py
 test_py/pose/__init__.py
 test_py/pose/human_pose.py
+test_py/qrcode/__init__.py
+test_py/qrcode/demo.py
 test_py/redis_py/__init__.py
 test_py/redis_py/knn_search.py
 test_py/redis_py/redis_client.py

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/test_py/aije/demo_labelme_crop_ailt.py

@@ -99,11 +99,12 @@ if __name__ == '__main__':
     """
     对labelme的数据目标进行裁剪，用于制作分类数据集
     """
+    # TODO 蓝色长条标签和鞋套是否相似
     target_name = ['身穿工作服', '未穿工作服',
                    '绝缘鞋', '脚穿绝缘鞋', '长筒靴', '脚穿长筒靴', '其他鞋', '脚穿其他鞋', '脚穿鞋套',
                    '手', '绝缘手套', '手穿绝缘手套', '棉纱手套', '手穿棉纱手套', '其他手套', '手穿其他手套',
                    '胶枪灯', '胶枪灯红色', '胶枪灯绿色',
-                   'PDA', 'PDA*', '手与*接触','电视柜台面'
+                   'PDA', 'PDA*', '手与*接触', '电视柜台面'
                    ]
     # '*长条标签'
     datasets = [

pybaseutils-2.4.7/test_py/demo2.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+"""
+    @Author : PKing
+    @E-mail : 390737991@qq.com
+    @Date   : 2022-12-31 11:37:30
+    @Brief  : https://blog.csdn.net/qdPython/article/details/121381363
+"""
+import os
+import cv2
+import random
+import types
+import torch
+import numbers
+from typing import List, Tuple, Dict
+import numpy as np
+from typing import Callable
+from PIL import Image
+from pybaseutils import image_utils, file_utils, text_utils, pandas_utils, json_utils, base64_utils
+import numpy as np
+import cv2
+import math
+
+import threading
+import time
+
+import numpy as np
+
+from test_py.aije.示例代码.MobileNetV2 import output
+
+
+def cal_contour_mean(image, contours):
+    """
+    计算指定轮廓内像素的平均值
+    :param image:
+    :param contours:
+    :return:
+    """
+    output = []
+    for c in contours:
+        mask = np.zeros(image.shape[:2], dtype=np.uint8)
+        cv2.drawContours(mask, [c], -1, 255, -1)  # 填充指定轮廓（白色）
+        mean = cv2.mean(image, mask=mask)  # 计算掩膜内像素的平均值
+        output.append(mean)
+    return output
+
+
+if __name__ == '__main__':
+    image_file = "/home/PKing/nasdata/dataset-dmai/AILT/ailt-line/dataset-20251216-test/images/20251216_191638_主视_13300.jpg"
+    image = cv2.imread(image_file)
+    centers = [[100, 200], [300, 400], [500, 600]]

{pybaseutils-2.4.5 → pybaseutils-2.4.7}/test_py/demo_video2frame.py

@@ -21,8 +21,9 @@ def video2frames_demo(root, out, vis=False):
         # video_utils.video2frames_similarity(video_file, out_dir=out, func=None, interval=20, thresh=0.3, vis=True)
         video_utils.video2frames(video_file, out_dir=out, task=None, interval=20, vis=vis, prefix="")
 
+
 if __name__ == "__main__":
-    root = "/home/PKing/edudata/dataset/AIJE/【TOP】技能人才系统_数据集管理/06-联通岗评/02-训练数据/联通入户布线_143_20251216_191638-打胶枪专题数据/20251216_191638_主视.mp4"
+    root = "/home/PKing/edudata/dataset/AIJE/【TOP】技能人才系统_数据集管理/06-联通岗评/02-训练数据/联通入户布线_156_20251219_180523-胶枪打胶专题数据/20251219_180523_主视.mp4"
     out = os.path.join(os.path.dirname(root), "frames")
     # out = os.path.join(root, "frames")
     video2frames_demo(root, out)

pybaseutils-2.4.7/test_py/qrcode/__init__.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-12-18 15:10:34
+# @Brief  :
+# --------------------------------------------------------
+"""