pybaseutils 2.2.8__tar.gz → 2.2.9__tar.gz

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.2.8
+Version: 2.2.9
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
@@ -137,4 +137,3 @@ base_utils(C++)
        DEBUG_IMSHOW("image", image);
        return 0;
    }
-

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils/__init__.py

@@ -5,4 +5,4 @@
     @Date   : 2019-05-07 17:40:27
 """
 
-__version__ = '2.2.8'
+__version__ = '2.2.9'

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils/cvutils/video_utils.py

@@ -299,12 +299,14 @@ def video_iterator(video_file: int or str, save_video: str or int = None, interv
     save_fps = max(kwargs.get("speed", 1) * fps // interval, 1)
     count = 0
     video_writer = None
+    use_fast = kwargs.get("use_fast", False)
     while True:
-        ret, frame = video_cap.read()
+        ret, frame = (False, None) if use_fast else video_cap.read()
         if count % interval == 0 and count >= start:
             # TODO 设置抽帧的位置，但某些格式视频容易出现问题
-            # if isinstance(video_file, str): video_cap.set(cv2.CAP_PROP_POS_FRAMES, count)
-            # ret, frame = video_cap.read()
+            if use_fast and isinstance(video_file, str):
+                video_cap.set(cv2.CAP_PROP_POS_FRAMES, count)
+                ret, frame = video_cap.read()
             if not ret or 0 < end < count or frame is None: break
             if task: frame = task(frame, **kwargs)
             data_info = {"frame": frame, "count": count, "w": w, "h": h, "fps": fps}

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils/dataloader/parser_image_text.py

@@ -16,7 +16,7 @@ import random
 import math
 import cv2
 from tqdm import tqdm
-from torch.utils.data import Dataset
+from pybaseutils.dataloader.base_dataset import Dataset, ConcatDataset
 from pybaseutils import image_utils, file_utils, json_utils
 from pybaseutils.dataloader import data_resample
 
@@ -138,32 +138,6 @@ class TextDataset(Dataset):
             item_list += data
         return item_list
 
-    def parser_classes(self, class_name):
-        """
-        class_dict = {class_name: i for i, class_name in enumerate(class_name)}
-        :param
-        :return:
-        """
-        if isinstance(class_name, str):
-            class_name = file_utils.read_data(class_name, split=None)
-        if isinstance(class_name, list) and len(class_name) > 0:
-            class_dict = {}
-            for i, name in enumerate(class_name):
-                name = name.split(",")
-                for n in name: class_dict[n] = i
-        elif isinstance(class_name, dict) and len(class_name) > 0:
-            class_dict = class_name
-            class_name = list(class_dict.keys())
-        else:
-            class_dict = None
-        if class_dict:
-            class_dict = json_utils.dict_sort_by_value(class_dict, reverse=False)
-            class_name = {}
-            for n, i in class_dict.items():
-                class_name[i] = "{},{}".format(class_name[i], n) if i in class_name else n
-            class_name = list(class_name.values())
-        return class_name, class_dict
-
     def check_item(self, item_list):
         """
         :param item_list:

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils/dataloader/parser_labelme.py

@@ -45,12 +45,13 @@ class LabelMeDataset(Dataset):
         :param shuffle:
         :param check: 当class_name=None且check=True,将自动获取所有class
         :param min_points: 当标注的轮廓点的个数小于min_points，会被剔除；负数不剔除
-        :param kwargs:
+        :param kwargs: read_image: 是否读取图片，否则image=None
         """
         super(LabelMeDataset, self).__init__()
         self.min_area = 1 / 1000  # 如果前景面积不足0.1%,则去除
         self.use_rgb = use_rgb
         self.min_points = min_points
+        self.kwargs = kwargs
         self.class_name, self.class_dict = self.parser_classes(class_name)
         parser = self.parser_paths(filename, data_root, anno_dir, image_dir)
         self.data_root, self.anno_dir, self.image_dir, self.image_ids = parser
@@ -154,8 +155,8 @@ class LabelMeDataset(Dataset):
         elif anno_dir and not image_ids:
             image_ids = self.get_file_list(anno_dir, postfix=["*.json", "*.xml"], basename=False)
             image_ids = [os.path.basename(f) for f in image_ids]
-        assert os.path.exists(image_dir), Exception("no image_dir:{}".format(image_dir))
-        assert os.path.exists(anno_dir), Exception("no anno_dir :{}".format(anno_dir))
+        assert isinstance(anno_dir, str) and os.path.exists(anno_dir), "no anno_dir :{}".format(anno_dir)
+        assert isinstance(image_dir, str) and os.path.exists(image_dir), "no image_dir:{}".format(image_dir)
         assert len(image_ids) > 0, f"image_ids is empty,image_dir={image_dir},anno_dir={anno_dir}"
         return data_root, anno_dir, image_dir, image_ids
 
@@ -167,13 +168,17 @@ class LabelMeDataset(Dataset):
         image_id = self.index2id(index)
         image_file, anno_file, image_id = self.get_image_anno_file(image_id)
         annotation, width, height = self.load_annotations(anno_file)
-        image = self.read_image(image_file, use_rgb=self.use_rgb)
-        shape = image.shape
-        data_info = self.parser_annotation(annotation, self.class_dict, shape, min_points=self.min_points,
-                                           unique=self.unique)
+        if self.kwargs.get("read_image", True):  # 是否读取图片
+            image = self.read_image(image_file, use_rgb=self.use_rgb)
+            shape = image.shape
+            size = [shape[1], shape[0]]
+        else:
+            image, shape, size = None, None, [width, height]
+        data_info = self.parser_annotation(annotation, self.class_dict, shape=shape,
+                                           min_points=self.min_points, unique=self.unique)
         # TODO dict(boxes, labels, points, groups, names, keypoints)
         data_info.update({"image": image, "image_file": image_file, "anno_file": anno_file,
-                          "size": [shape[1], shape[0]]})
+                          "size": size})
         return data_info
 
     @staticmethod
@@ -448,7 +453,7 @@ def parser_labelme(anno_file, class_dict={}, shape=None):
 def draw_keypoints_image(image, boxes=[], keypoints=[], thickness=1, vis_id=False):
     """绘制keypoints"""
     h, w = image.shape[:2]
-    if len(keypoints)== 0: return image
+    if len(keypoints) == 0: return image
     if len(boxes) == 0: boxes = [(0, 0, w, h)] * len(keypoints)
     from pybaseutils.pose import bones_utils
     target_bones = bones_utils.get_target_bones("coco_person")
@@ -480,7 +485,6 @@ if __name__ == "__main__":
     from pybaseutils.converter import build_labelme
 
     anno_dir = "/home/PKing/Downloads/sample/images"
-    anno_dir = "/home/PKing/Downloads/冲击试验/video-ok/NVR_ch2_main_20241121093213_20241121100115/images"
     # anno_dir = "/home/PKing/Downloads/冲击试验/sample/images"
     names = None
     dataset = LabelMeDatasets(filename=None,

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils/file_utils.py

@@ -22,6 +22,7 @@ import argparse
 import itertools
 from datetime import datetime
 from tqdm import tqdm
+from pybaseutils import text_utils
 
 IMG_POSTFIX = ['*.jpg', '*.jpeg', '*.png', '*.tif', "*.JPG", "*.bmp"]
 VIDEO_POSTFIX = ['*.mp4', '*.avi', '*.mov', "*.flv", "*.dav"]
@@ -793,7 +794,7 @@ def get_files_lists(file_dir, postfix=IMG_POSTFIX, subname="", shuffle=False, su
     return file_list
 
 
-def get_files_list(file_dir, prefix="", postfix=None, basename=False, sub=False):
+def get_files_list_v1(file_dir, prefix="", postfix=None, basename=False, sub=False):
     """
     获得file_dir目录下，后缀名为postfix所有文件列表，包括子目录所有文件
     :param file_dir:
@@ -822,6 +823,25 @@ def get_files_list(file_dir, prefix="", postfix=None, basename=False, sub=False)
     return file_list
 
 
+def get_files_list(file_dir, prefix="", postfix=None, basename=False, sub=False):
+    """
+    获得file_dir目录下，后缀名为postfix所有文件列表，包括子目录所有文件
+    :param file_dir:
+    :param prefix: 前缀
+    :param postfix: 后缀
+    :param basename: 返回的列表是文件名（True），还是文件的完整路径(False)
+    :param sub: 是否去除根路径
+    :return:
+    """
+    file_list = get_all_files(file_dir)
+    file_list = text_utils.find_match_texts(file_list, pattern=[prefix], org=True) if prefix else file_list
+    file_list = text_utils.find_match_texts(file_list, pattern=postfix, org=True) if postfix else file_list
+    file_list.sort()
+    file_list = get_basename(file_list) if basename else file_list
+    if sub: file_list = get_sub_list(file_list, dirname=file_dir)
+    return file_list
+
+
 def get_files_list_v2(file_dir, prefix="", postfix=None, basename=False):
     """
     获取file_dir目录下，所有文本路径，但不包括子目录文件

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils/time_utils.py

@@ -51,7 +51,8 @@ def performance(tag=""):
                                                                                            content["avg"],
                                                                                            content["total"],
                                                                                            content["count"])
-            print("{} call {} elapsed: {}".format(tag, func.__name__, elapsed))
+            tag_ = f"{tag} " if tag else ""
+            print("{:20s}{:20s} elapsed: {}".format(tag_, func.__name__, elapsed))
             return result
 
         return wrapper
@@ -78,37 +79,38 @@ class Performance(object):
                                                                                        content["avg"],
                                                                                        content["total"],
                                                                                        content["count"])
-        print("{} elapsed: {}".format(self.tag, elapsed))
+        tag_ = f"{self.tag} " if self.tag else ""
+        print("{:20s}elapsed: {}\t".format(tag_, elapsed))
 
     def task(self):
         pass
 
 
-@performance("")
+@performance("test1")
 def targe_func1():
     time.sleep(1)
 
 
-@performance("")
+@performance("test111111")
 def targe_func2():
     time.sleep(0.5)
 
 
 def targe_func3():
-    with Performance("targe_func3") as p:
+    with Performance("test222") as p:
         time.sleep(1)
 
 
 def targe_func4():
-    with Performance("targe_func4") as p:
+    with Performance("test22222222") as p:
         time.sleep(0.5)
 
 
 def targe_func():
     targe_func1()
     targe_func2()
-    targe_func3()
-    targe_func4()
+    # targe_func3()
+    # targe_func4()
 
 
 if __name__ == '__main__':

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.2.8
+Version: 2.2.9
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
@@ -137,4 +137,3 @@ base_utils(C++)
        DEBUG_IMSHOW("image", image);
        return 0;
    }
-

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/pybaseutils.egg-info/SOURCES.txt

@@ -190,6 +190,9 @@ test_py/audio/main_read.py
 test_py/audio/segment.py
 test_py/audio/speechbrain_asr_indoor_prod.py
 test_py/audio/speechbrain_demo.py
+test_py/camera/__init__.py
+test_py/camera/demo.py
+test_py/camera/main.py
 test_py/captcha/__init__.py
 test_py/captcha/demo.py
 test_py/converter/AffectNet.py

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/test_py/aije/demo_labelme_crop.py

@@ -64,7 +64,7 @@ if __name__ == "__main__":
           '绝缘手套', '绝缘鞋', '脚扣', '螺丝', '身穿工作服', '遮拦杆', '铁架', '铝扎线']
     室外: []
     """
-    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/test-v1/aije-action-train-v02/json"
+    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")
     dataset = parser_labelme.LabelMeDatasets(filename=None,
@@ -77,8 +77,8 @@ if __name__ == "__main__":
                                              shuffle=False)
     print("have num:{}".format(len(dataset)))
     class_name = dataset.class_name
-    scale = [1.5,1.5]
-    flag = str(scale[0]).replace(".", "p")
+    scale = [1.0, 1.0]
+    # flag = str(scale[0]).replace(".", "p")
     flag = None
     # scale = None
     for i in tqdm(range(len(dataset))):

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/test_py/aije/demo_video_aije.py

@@ -84,15 +84,15 @@ def video2frames_demo(root, out, prefix="", thresh=0.2):
     for video_file in tqdm(files):
         print(video_file)
         name = os.path.basename(video_file).split(".")[0]
-        if name in thresh_dict:
-            thresh = thresh_dict[name]
-        prefix_ = "{}_{}".format(prefix, os.path.basename(os.path.dirname(video_file)))
+        thresh = thresh_dict.get(name, 0.5)
+        prefix_ = os.path.basename(os.path.dirname(video_file))
+        if prefix: prefix_ = "{}_{}".format(prefix, prefix_)
         video2frames_similarity(video_file, out_dir=out, func=None, interval=20,
                                 thresh=thresh, prefix=prefix_, vis=True)
 
 
 if __name__ == "__main__":
-    root = "/home/PKing/nasdata/dataset-dmai/AIJE/技能人才系统_数据集管理/归档/20240607_数企_jinquan采集的多样性样本-train/shuqi_20240607yanshi111346/shuqi_20240607yanshi111346_right.mp4"
-    prefix = "数企"
+    root = "/home/PKing/nasdata/dataset-dmai/AIJE/技能人才系统_数据集管理/01-江门四维数据/01-37道考题视频/江门四维2024-12-06-检查帽撑/video"
+    prefix = ""
     out = root + "-train"
     video2frames_demo(root, out, prefix=prefix)

pybaseutils-2.2.9/test_py/camera/__init__.py

@@ -0,0 +1,7 @@
+# -*-coding: utf-8 -*-
+"""
+    @Author : PKing
+    @E-mail : 
+    @Date   : 2024-12-17 11:45:18
+    @Brief  :
+"""

pybaseutils-2.2.9/test_py/camera/demo.py

@@ -0,0 +1,89 @@
+# -*-coding: utf-8 -*-
+"""
+    @Author : PKing
+    @E-mail : 
+    @Date   : 2024-12-17 15:02:18
+    @Brief  :
+"""
+import cv2
+import time
+from datetime import datetime, timedelta
+
+
+def display_hikvision_stream(rtsp_url):
+    print(f"正在连接摄像头: {rtsp_url}")
+    cap = cv2.VideoCapture(rtsp_url)
+
+    if not cap.isOpened():
+        print("无法连接到摄像头")
+        return
+
+    # 获取视频信息
+    fourcc = int(cap.get(cv2.CAP_PROP_FOURCC))
+    codec = "".join([chr((fourcc >> 8 * i) & 0xFF) for i in range(4)])
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+    print("视频流信息:")
+    print(f"编码格式: {codec}")
+    print(f"分辨率: {width}x{height}")
+    print(f"帧率: {fps}")
+
+    # 获取起始时间
+    start_time = datetime.now()
+    start_timestamp = time.time() * 1000  # 转换为毫秒
+
+    try:
+        while True:
+            ret, frame = cap.read()
+            if ret:
+                # 获取时间戳信息
+                timestamp_ms = cap.get(cv2.CAP_PROP_POS_MSEC)
+                current_time = datetime.now()
+
+                # 计算摄像头实际时间
+                # if timestamp_ms == 0:  # 如果摄像头没有返回有效时间戳
+                #     elapsed_ms = (time.time() * 1000) - start_timestamp
+                #     camera_datetime = start_time + timedelta(milliseconds=elapsed_ms)
+                # else:
+                camera_datetime = start_time + timedelta(milliseconds=timestamp_ms)
+
+                # 格式化时间字符串
+                camera_time_str = camera_datetime.strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
+                system_time_str = current_time.strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
+
+                # 在图像上显示时间戳
+                cv2.putText(frame, f"Camera Time: {camera_time_str}", (10, 30),
+                            cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+                cv2.putText(frame, f"System Time: {system_time_str}", (10, 70),
+                            cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+
+                # 显示视频帧
+                frame = cv2.resize(frame, (1280, 720))
+                cv2.imshow('Camera', frame)
+
+                # 打印时间戳信息到控制台
+                print(f"\r摄像头时间: {camera_time_str} | 系统时间: {system_time_str}",
+                      end="", flush=True)
+
+                # 按'q'键退出
+                if cv2.waitKey(1) & 0xFF == ord('q'):
+                    break
+            else:
+                print("\n视频帧获取失败，正在重新连接...")
+                cap.release()
+                time.sleep(1)
+                cap = cv2.VideoCapture(rtsp_url)
+
+    finally:
+        cap.release()
+        cv2.destroyAllWindows()
+        print("\n程序已退出")
+
+
+if __name__ == "__main__":
+    camera_url = "rtsp://admin:C2332416@192.168.2.35/Streaming/Channels/1"
+    # display_hikvision_stream(camera_url)
+    v = (100, 100) + (100, 100)
+    print(v)

pybaseutils-2.2.9/test_py/camera/main.py

@@ -0,0 +1,147 @@
+import os
+import cv2
+import time
+import requests
+import xmltodict
+from datetime import datetime
+from requests.auth import HTTPDigestAuth
+
+
+class CameraDemo():
+    def __init__(self, ip, username, password):
+        """
+        按'q'键退出
+        :param ip: IP摄像头IP地址
+        :param username: 用户
+        :param password: 密码
+        """
+        self.ip = ip
+        self.username = username
+        self.password = password
+
+    def get_camera_time(self):
+        """获取海康摄像头的时间
+        urls = [
+                f'http://{ip}/ISAPI/System/time',
+                f'http://{ip}/ISAPI/System/status',
+                f'http://{ip}/ISAPI/System/deviceInfo'
+            ]
+        """
+        t = "unknown camera time"
+        try:
+            # 尝试不同的 ISAPI 接口
+            url = f'http://{self.ip}/ISAPI/System/time'
+            response = requests.get(url, auth=HTTPDigestAuth(self.username, self.password), verify=False, timeout=5)
+            content = response.content
+            content = xmltodict.parse(content)  # 将XML格式json格式
+            if response.status_code == 200:
+                t = content['Time']['localTime']
+                t = t[:-len('+08:00')]  # 去除+08:00字符
+            else:
+                print(f"code: {response.status_code},content={response.text}")
+        except Exception as e:
+            print(f"Error: {e}")
+        return t
+
+    def get_camera_info(self, cap):
+        """
+        获取视频信息
+        :param cap:
+        :return:
+        """
+        fourcc = int(cap.get(cv2.CAP_PROP_FOURCC))
+        codec = "".join([chr((fourcc >> 8 * i) & 0xFF) for i in range(4)])
+        fps = cap.get(cv2.CAP_PROP_FPS)
+        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        print(f"视频流信息, 编码格式:{codec}, 分辨率:{width}x{height}, 帧率:{fps}")
+        return width, height, fps
+
+    def draw_image_info(self, frame, info: str, point, color=(0, 255, 0), fontScale=1.2, thickness=2):
+        cv2.putText(frame, info, point, cv2.FONT_HERSHEY_SIMPLEX, fontScale, color, thickness)
+        return frame
+
+    @staticmethod
+    def get_video_writer(video_file, width, height, fps):
+        """
+        获得视频存储对象
+        :param video_file: 输出保存视频的文件路径
+        :param width:   图像分辨率width
+        :param height:  图像分辨率height
+        :param fps:  设置视频播放帧率
+        :return:
+        """
+        if not os.path.exists(os.path.dirname(video_file)):
+            os.makedirs(os.path.dirname(video_file))
+        fourcc = cv2.VideoWriter_fourcc(*'XVID')
+        frameSize = (int(width), int(height))
+        video_writer = cv2.VideoWriter(video_file, fourcc, int(fps), frameSize)
+        print("save video:width:{},height:{},fps:{}".format(width, height, fps))
+        return video_writer
+
+    def capture(self, title="camera", save_video="./camera.avi"):
+        format = '%Y-%m-%dT%H:%M:%S'  # 海康摄像头默认格式
+        video_url = f"rtsp://{self.username}:{self.password}@{self.ip}/Streaming/Channels/1"
+        print(f"正在连接摄像头: {video_url}")
+        # 创建视频流连接
+        cap = cv2.VideoCapture(video_url)
+        assert cap.isOpened(), f"无法连接到摄像头:{video_url}"
+        W, H, Fps = self.get_camera_info(cap)
+        video_writer = self.get_video_writer(save_video, width=W, height=H, fps=Fps)
+        # TODO 获得启动摄像头的时间
+        cam_start_time = self.get_camera_time()
+        cam_start_timestamp = time.mktime(time.strptime(cam_start_time, format))
+        reconnection = 0
+        count = 0
+        point = (W - 750, 150)
+        while True:
+            ret, frame = cap.read()
+            count += 1
+            if ret:
+                # TODO 获取当前系统时间
+                sys_timestamp = time.time()
+                sys_time = datetime.fromtimestamp(sys_timestamp).strftime(format)
+                # TODO 通过接口获得摄像头设备时间
+                cam_time = self.get_camera_time()
+                cam_timestamp = time.mktime(time.strptime(cam_time, format))
+                # TODO 通过视频解码估计视频流生成时间
+                time.sleep(0.1)  # TODO 模拟网络延时
+                vdo_offset = cap.get(cv2.CAP_PROP_POS_MSEC) / 1000  # 获得视频偏移量
+                vdo_timestamp = cam_start_timestamp + vdo_offset  # 设备启动时间+视频偏移量得到当前帧的到达时间
+                vdo_time = datetime.fromtimestamp(vdo_timestamp).strftime(format)
+                # 在图像上显示时间戳
+                sys_info = f"System Time: {sys_time}"
+                cam_info = f"Camera Time: {cam_time}"
+                vdo_info = f"Video  Time: {vdo_time}"
+                # 打印时间戳信息到控制台
+                print(f"\r{cam_info} | {sys_info}| {vdo_info}", end="", flush=True)
+                frame = self.draw_image_info(frame, info=vdo_info, point=(point[0], point[1]))
+                frame = self.draw_image_info(frame, info=cam_info, point=(point[0], point[1] + 50))
+                frame = self.draw_image_info(frame, info=sys_info, point=(point[0], point[1] + 100))
+                # 显示视频帧
+                cv2.namedWindow(title, flags=cv2.WINDOW_NORMAL)
+                cv2.imshow(title, frame)
+                # 保存视频
+                video_writer.write(frame)
+                # 按'q'键退出
+                if cv2.waitKey(10) & 0xFF == ord('q'):
+                    break
+            else:
+                cap.release()
+                reconnection += 1
+                print(f"视频帧获取失败，尝试第{reconnection}次重新连接...")
+                time.sleep(10)
+                cap = cv2.VideoCapture(video_url)
+        cap.release()
+        video_writer.release()
+        cv2.destroyAllWindows()
+        print("\n程序已退出")
+
+
+if __name__ == "__main__":
+    # 摄像头信息，按'q'键退出程序
+    ip = "192.168.2.35"
+    username = "admin"
+    password = "C2332416"
+    c = CameraDemo(ip, username, password)
+    c.capture()

pybaseutils-2.2.9/test_py/demo1.py

@@ -0,0 +1,17 @@
+# -*-coding: utf-8 -*-
+"""
+    @Author : PKing
+    @E-mail :
+    @Date   : 2024-02-05 18:19:18
+    @Brief  :
+"""
+
+from pybaseutils import file_utils, image_utils
+
+if __name__ == '__main__':
+    root = "/home/PKing/Downloads/data1"
+    files1 = file_utils.get_files_list_v1(root, postfix=["*.jpg"])
+    files2 = file_utils.get_files_list(root, postfix=["*.jpg", "*.png"])
+    [print(f) for f in files1]
+    print("----" * 10)
+    [print(f) for f in files2]

{pybaseutils-2.2.8 → pybaseutils-2.2.9}/test_py/demo_lableme_vis.py

@@ -26,7 +26,7 @@ if __name__ == "__main__":
     anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-indoor-det/dataset-test/json"
     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-v2-det/dataset-v31/images"
+    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/train-v2/aije-action-train-v30/images"
     names = None
     # names = ['身穿工作服,未穿工作服', '手,手穿绝缘手套,手穿棉纱手套,手穿其他手套']
     dataset = parser_labelme.LabelMeDatasets(filename=None,

pybaseutils-2.2.8/test_py/demo1.py

@@ -1,55 +0,0 @@
-# -*-coding: utf-8 -*-
-"""
-    @Author : PKing
-    @E-mail :
-    @Date   : 2024-02-05 18:19:18
-    @Brief  :
-"""
-import os
-import cv2
-import numpy as np
-import random
-import json
-from PIL import Image
-from typing import Dict
-from tqdm import tqdm
-from pybaseutils import image_utils, file_utils, json_utils, base64_utils, time_utils
-from pybaseutils.cvutils import video_utils
-from pybaseutils.transforms import transform_utils
-from pybaseutils.converter import build_voc, build_labelme
-from pybaseutils.dataloader import parser_labelme
-import xmltodict
-
-import torch
-import torch.nn as nn
-
-
-class CMDLoss(nn.Module):
-    """计算两个分布的中心矩差差异的Loss函数"""
-
-    def __init__(self, num_channels, use_gpu=True):
-        super(CMDLoss, self).__init__()
-        self.use_gpu = use_gpu
-        self.num_channels = num_channels
-        self.register_buffer('one', torch.tensor(1.0))
-
-    def forward(self, input, target):
-        # 计算均值
-        input_mean = torch.mean(input, dim=[2, 3], keepdim=True)
-        target_mean = torch.mean(target, dim=[2, 3], keepdim=True)
-
-        # 计算方差
-        input_var = torch.mean((input - input_mean) ** 2, dim=[2, 3], keepdim=True)
-        target_var = torch.mean((target - target_mean) ** 2, dim=[2, 3], keepdim=True)
-
-        # 计算协方差
-        input_mean_expand = input_mean.expand_as(input)
-        target_mean_expand = target_mean.expand_as(target)
-        cov = torch.mean((input - input_mean_expand) * (target - target_mean_expand), dim=[2, 3], keepdim=True)
-
-        # 计算CMD
-        c = torch.mean(self.one / self.num_channels * cov, dim=[2, 3], keepdim=True)
-        d = torch.mean(self.one / self.num_channels * (input_var - target_var), dim=[2, 3], keepdim=True)
-        cmd_loss = c / (d + 1e-5)
-
-        return cmd_loss