pybaseutils 2.3.7__tar.gz → 2.4.0__tar.gz

{pybaseutils-2.3.7 → pybaseutils-2.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.3.7
+Version: 2.4.0
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan

{pybaseutils-2.3.7 → pybaseutils-2.4.0}/pybaseutils/__init__.py

@@ -8,4 +8,4 @@
 # --------------------------------------------------------
 """
 
-__version__ = '2.3.7'
+__version__ = '2.4.0'

{pybaseutils-2.3.7 → pybaseutils-2.4.0}/pybaseutils/file_utils.py

@@ -188,31 +188,19 @@ class WriterTXT(object):
 
 def parser_classes(class_name):
     """
-    class_dict = {class_name: i for i, class_name in enumerate(class_name)}
-    :param class_name:
-                str : class file
-                list: ["face","person"]
-                dict: 可以自定义label的id{'BACKGROUND': 0, 'person': 1, 'person_up': 1, 'person_down': 1}
+    TODO 很多模型使用parser_classes，不要做任何修改！！！
     :return:
     """
     if isinstance(class_name, str):
         class_name = 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:
+    elif isinstance(class_name, numbers.Number):
+        class_name = [i for i in range(int(class_name))]
+    if isinstance(class_name, list):
+        class_dict = {class_name: i for i, class_name in enumerate(class_name)}
+    elif isinstance(class_name, dict):
         class_dict = class_name
-        class_name = list(class_dict.keys())
     else:
         class_dict = None
-    if class_dict:
-        # class_dict = json_utils.dict_sort(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
 
 

{pybaseutils-2.3.7 → pybaseutils-2.4.0}/pybaseutils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.3.7
+Version: 2.4.0
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan

pybaseutils-2.4.0/test_py/demo4.py

@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail : 
+# @Date   : 2025-07-08 14:10:15
+# @Brief  : 转换labelme标注数据为voc格式
+# --------------------------------------------------------
+"""
+import os
+import numpy as np
+from pybaseutils.converter import convert_labelme2voc
+from pybaseutils import time_utils
+
+import numpy as np
+from collections import deque
+from scipy.signal import convolve
+
+
+def get_valid_range(data, v=0):
+    """
+    获得值大于v的最大区间
+    :param data:
+    :param v:
+    :return: [index0,index1) 左包含，右不包含
+    """
+    if not isinstance(data, np.ndarray): data = np.array(data)
+    index = np.where(data > v)[0]
+    ranges = (0, 0)
+    if len(index) > 0:
+        ranges = (index[0], index[-1] + 1)
+    return ranges
+
+
+def get_window_sum(score, times, stage_time, ksize=5):
+    """
+    :param score:
+    :param ksize:
+    :return:
+    """
+    score = np.array(score)
+    # i1, i2 = get_valid_range(score, v=0)
+    ksize = np.ones(ksize)  # 创建滑动窗口核
+    # 使用卷积计算滑动窗口和,'same'模式保持输出长度与输入相同
+    result = convolve(score, ksize, mode='same')
+    # result = result * (score != 0)  # 只有当data[i]不为0时才保留窗口和
+    n = len(result)
+    area = 0
+    time = stage_time
+    i = 0
+    while i < n:
+        if result[i] > 0:
+            c0 = i
+            while i < n and result[i] > 0:  # 找到事件的开始和结束
+                i += 1
+            c1 = i
+            wsum = sum(result[c0:c1])  # [c0,c1), 左包含，右不包含
+            print(wsum, (c0, c1))
+            if wsum > area:
+                time = (times[c0], times[c1 - 1])
+                area = wsum
+        i += 1
+    print("time={}".format(time))
+    return time, result
+
+
+if __name__ == "__main__":
+    score = [0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 3, 2, 0, 0, 1]
+    score = np.asarray(score, dtype=np.float32)
+    times = np.arange(len(score))
+    stage_time = (0, len(score) - 1)
+    ksize = 2
+    print("function:", get_window_sum(score, times, stage_time=stage_time, ksize=ksize))
+    print("原始数据  :", score)

pybaseutils-2.3.7/test_py/demo4.py

@@ -1,19 +0,0 @@
-# -*- coding: utf-8 -*-
-"""
-# --------------------------------------------------------
-# @Author : Pan
-# @E-mail : 
-# @Date   : 2025-07-08 14:10:15
-# @Brief  : 转换labelme标注数据为voc格式
-# --------------------------------------------------------
-"""
-import os
-from pybaseutils.converter import convert_labelme2voc
-if __name__ == "__main__":
-    # 将image_dir修改你labelme标注数据的根目录
-    image_dir = "/home/PKing/nasdata/tmp/tmp/pen/笔尖指尖标注方法/JPEGImages"
-    json_dir  = image_dir
-    out_root = os.path.dirname(json_dir)
-    class_dict = {}
-    lm = convert_labelme2voc.Labelme2VOC(image_dir, json_dir)
-    lm.build_dataset(out_root, class_dict=class_dict, vis=False, crop=False)