pybaseutils 2.1.2__tar.gz → 2.1.3__tar.gz

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/PKG-INFO

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

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/pybaseutils/__init__.py

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

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/pybaseutils/cvutils/video_utils.py

@@ -13,7 +13,7 @@ from tqdm import tqdm
 from pybaseutils import image_utils, file_utils
 from pybaseutils.cvutils import monitor
 
-VIDEO_POSTFIX = ['*.mp4', '*.avi']
+VIDEO_POSTFIX = ['*.mp4', '*.avi', '*.mov']
 
 get_video_capture = image_utils.get_video_capture
 get_video_info = image_utils.get_video_info

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/pybaseutils/file_utils.py

@@ -24,7 +24,7 @@ from datetime import datetime
 from tqdm import tqdm
 
 IMG_POSTFIX = ['*.jpg', '*.jpeg', '*.png', '*.tif', "*.JPG", "*.bmp"]
-VIDEO_POSTFIX = ['*.mp4', '*.avi']
+VIDEO_POSTFIX = ['*.mp4', '*.avi', '*.mov']
 
 
 def str2bool(v):

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/pybaseutils.egg-info/PKG-INFO

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

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/pybaseutils.egg-info/SOURCES.txt

@@ -154,6 +154,7 @@ test_py/demo_voc_vis.py
 test_py/demo_word_similar.py
 test_py/demo_worker1.py
 test_py/demo_worker2.py
+test_py/get_file_list.py
 test_py/kafka_worker.py
 test_py/men_tracemalloc.py
 test_py/performance.py

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/test_py/aije/convert_cvat2labelme.py

@@ -14,8 +14,8 @@ from pybaseutils.converter import build_voc, build_labelme, convert_cvat2labelme
 from pybaseutils import file_utils, image_utils, json_utils
 
 if __name__ == "__main__":
-    image_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/aije-action-train-v01/JPEGImages"
+    image_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-v2-det/dataset-v25/JPEGImages"
     anno_dir = os.path.join(os.path.dirname(image_dir), "xml")
     json_dir = os.path.join(os.path.dirname(image_dir), "json")
     convert_cvat2labelme.convert_cvat2labelme(image_dir=image_dir, anno_dir=anno_dir, out_dir=json_dir,
-                                              thickness=1, fontScale=2.0, vis=False)
+                                              thickness=1, fontScale=2.0, vis=True)

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/test_py/aije/demo_labelme_crop.py

@@ -64,14 +64,8 @@ if __name__ == "__main__":
           '绝缘手套', '绝缘鞋', '脚扣', '螺丝', '身穿工作服', '遮拦杆', '铁架', '铝扎线']
     室外: []
     """
-    # anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-indoor-det/dataset-v9/json"
-    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-indoor-det/dataset-v0/json"
-    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-indoor-det-fix/dataset-v12/json"
-    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-det/aije-action-train-v20/json"
-    # anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-indoor-det/dataset-v10/json"
-    # anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-person-coco/json"
-    # class_name = ['手', '绝缘手套', '手穿绝缘手套', '棉纱手套', '手穿棉纱手套']
-    class_name = None  # 室内
+    anno_dir = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-v2-det/dataset-v25/json"
+    class_name = None  # 室内\
     out_dir = os.path.join(os.path.dirname(anno_dir), "crops")
     dataset = parser_labelme.LabelMeDatasets(filename=None,
                                              data_root=None,
@@ -83,7 +77,7 @@ if __name__ == "__main__":
                                              shuffle=False)
     print("have num:{}".format(len(dataset)))
     class_name = dataset.class_name
-    scale = [2.0, 2.0]
+    scale = [1.0, 1.0]
     flag = str(scale[0]).replace(".", "p")
     flag = None
     # scale = None

{pybaseutils-2.1.2 → pybaseutils-2.1.3}/test_py/demo1.py

@@ -10,6 +10,7 @@ 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
@@ -49,9 +50,3 @@ class Promote():
 
 if __name__ == '__main__':
     image_file = "/home/PKing/nasdata/dataset-dmai/AIJE/dataset/aije-action-cvlm-v2/第3批数据/aije-action-train-v22/江门_检查工器具_20240910_172411_6606_0000.jpg"
-    boxe1 = [[600, 300, 850, 800]]
-    image = image_utils.read_image(image_file)
-    boxe2 = image_utils.get_square_boxes(boxe1, use_max=True, use_mean=True)
-    image = image_utils.draw_image_boxes(image, boxe1, color=(255, 0, 0))
-    image = image_utils.draw_image_boxes(image, boxe2, color=(0, 255, 0))
-    image_utils.cv_show_image("image", image)

pybaseutils-2.1.3/test_py/get_file_list.py

@@ -0,0 +1,36 @@
+# -*-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 numpy as np
+from typing import Callable
+from pybaseutils import image_utils, file_utils, text_utils
+from pybaseutils.cvutils import video_utils
+import cv2
+import re
+
+
+def get_file_list_dataset(image_dir, subs=[]):
+    files = file_utils.get_files_lists(image_dir, sub=True)
+    sub = os.path.basename(image_dir)
+    content = []
+    for path in files:
+        label = path.split("/")[0]
+        if label not in subs: continue
+        text = [os.path.join(sub, path), label]
+        content.append(text)
+    content = sorted(content)
+    filename = os.path.join(os.path.dirname(image_dir), f"{sub}.txt")
+    file_utils.write_data(filename, content, split=",")
+
+
+if __name__ == '__main__':
+    image_dir = "/home/PKing/nasdata/tmp/tmp/RealFakeFace/anti-spoofing-images-v1/test"
+    get_file_list_dataset(image_dir)