pybaseutils 2.3.2__tar.gz → 2.3.4__tar.gz

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/PKG-INFO

@@ -1,18 +1,13 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.1
 Name: pybaseutils
-Version: 2.3.2
+Version: 2.3.4
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan
 Author-email: 390737991@qq.com
 License: MIT
+Platform: UNKNOWN
 License-File: LICENCE
-Dynamic: author
-Dynamic: author-email
-Dynamic: description
-Dynamic: home-page
-Dynamic: license
-Dynamic: summary
 
 base-utils
 ==========
@@ -147,3 +142,5 @@ base_utils(C++)
        DEBUG_IMSHOW("image", image);
        return 0;
    }
+
+

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/__init__.py

@@ -8,4 +8,4 @@
 # --------------------------------------------------------
 """
 
-__version__ = '2.3.2'
+__version__ = '2.3.4'

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/base64_utils.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-03-22 09:11:35
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2022-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import sys
 import os

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/batch_utils.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-06-06 14:34:38
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2022-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 from typing import Tuple, List, Dict
 

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/build_cvat.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-09-05 19:48:52
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/build_labelme.py

@@ -15,13 +15,13 @@ from pybaseutils import image_utils, file_utils, json_utils
 def save_labelme(out_root, image_file, points, names, class_dict, image=None, prefix="",
                  index=0, vis=False, delay=0):
     """
-    :param out_root: 输入根目录
+    :param out_root: 输出根目录
     :param image_file: 图片路径
     :param points: 目标轮廓
     :param names:  目标名称
     :param class_dict: 需要映射的类别
     :param image: 图像
-    :param prefix: 前缀，如果提供，则重新名称
+    :param prefix: 前缀，如果提供，则重新命名
     :param index: 提供前缀需要重新名称
     :return:
     """
@@ -32,7 +32,7 @@ def save_labelme(out_root, image_file, points, names, class_dict, image=None, pr
         print("Error: image is None,image_file={},names={}".format(image_file, names))
         return
     if vis:
-        image = image_utils.draw_image_contours(image, points, texts=names)
+        image = image_utils.draw_image_contours(image, points, texts=names, alpha=0.3)
         image = image_utils.show_image("image", image, delay=delay)
     image_name = os.path.basename(image_file)
     image_id, postfix = file_utils.split_postfix(image_name)

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/build_voc.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-09-05 19:48:52
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 
 import cv2

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/concat_coco.py

@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
 """
 # --------------------------------------------------------
-# @Project:
-# @Author : panjq
-# @Date   : 2020-02-12 18:28:16
-# @url    :
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
 # --------------------------------------------------------
 """
 

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_coco2labelme.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_coco2voc.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_cvat2labelme.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_labelme2cvat.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_labelme2labelme.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np
@@ -28,7 +30,7 @@ def convert_labelme2labelme(anno_dir,
     :param out_root:  输出labelme根目录
     :param class_name: 需要选择的类别，None表示全部
     :param class_dict: 类别映射
-    :param prefix: 提供文件名前缀，则重新进行重新命令
+    :param prefix: 提供文件名前缀，则进行重新命名
     :param max_num: 最多转换样本个数
     """
     if anno_dir and not out_root: out_root = os.path.join(os.path.dirname(anno_dir), "labelme")

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_labelme2voc.py

@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
 """
 # --------------------------------------------------------
-# @Project: AlphaPose
 # @Author : Pan
-# @E-mail : 390737991@qq.com
-# @Date   : 2020-02-14 09:15:52
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
 # --------------------------------------------------------
 """
 import sys

pybaseutils-2.3.4/pybaseutils/converter/convert_labelme2yolo.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
+"""
+
+import os
+import numpy as np
+import cv2
+from tqdm import tqdm
+from pybaseutils.dataloader import parser_labelme, parser_yolo
+from pybaseutils import file_utils, image_utils
+
+
+def convert_labelme2yolo(anno_dir, out_root=None, class_name=None, use_seg=False, max_num=-1, vis=True, **kwargs):
+    """
+    将labelme格式转换为YOLO数据格式
+    :param anno_dir:  输入labelme根目录(图片和json文件必须放在同一目录)
+    :param out_root:  输出YOLO数据格根目录
+    :param class_name: 需要选择的类别，None表示全部
+    :param use_seg: 数据格式，True是YOLO实例分割数据格式 [class_index, cx, cy, w,  h]
+                            False是YOLO目标检测格式 [class_index, x1, y1, x2, y2, x3, y3, x4, y4,....]
+    :param max_num: 最多转换样本个数
+    :param vis: 可视化标注效果
+    """
+    if anno_dir and not out_root: out_root = os.path.join(os.path.dirname(anno_dir), "yolo")
+    dataset = parser_labelme.LabelMeDatasets(filename=None,
+                                             data_root=None,
+                                             anno_dir=anno_dir,
+                                             image_dir=None,
+                                             class_name=class_name,
+                                             check=False,
+                                             phase="val",
+                                             shuffle=False)
+    print("have num:{}".format(len(dataset)))
+    nums = min(len(dataset), max_num) if max_num > 0 else len(dataset)
+    for i in tqdm(range(nums)):
+        data_info = dataset.__getitem__(i)
+        labels, points, boxes = data_info["labels"], data_info["points"], data_info["boxes"]
+        image, image_file = data_info["image"], data_info["image_file"]
+        parser_yolo.save_yolo(out_root, image_file=image_file, boxes=boxes, points=points, labels=labels,
+                              use_seg=use_seg, image=image, vis=vis)
+    file_utils.write_data(os.path.join(out_root, "class_name.txt"), dataset.class_name)
+
+
+if __name__ == "__main__":
+    class_name = ['AngelFish', 'BlueTang', 'ButterflyFish', 'ClownFish', 'GoldFish', 'Gourami', 'MorishIdol',
+                  'PlatyFish', 'RibbonedSweetlips', 'ThreeStripedDamselfish', 'YellowCichlid', 'YellowTang',
+                  'ZebraFish']
+    anno_dir = "/home/PKing/nasdata/tmp/tmp/Fish/train/images"
+    convert_labelme2yolo(anno_dir, class_name=class_name, use_seg=False, vis=False)

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_voc2coco.py

@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
 """
 # --------------------------------------------------------
-# @Project:
-# @Author : panjq
-# @Date   : 2020-02-12 18:28:16
-# @url    :
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
 # --------------------------------------------------------
 """
 

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_voc2labelme.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_voc2voc.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_voc2yolo.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

pybaseutils-2.3.4/pybaseutils/converter/convert_yolo2labelme.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
+"""
+import os
+import numpy as np
+import cv2
+from tqdm import tqdm
+from pybaseutils.converter import build_labelme
+from pybaseutils.dataloader import parser_yolo
+from pybaseutils import file_utils, image_utils
+
+
+def convert_yolo2labelme(data_root,
+                         out_root=None,
+                         class_name=None,
+                         class_dict={},
+                         prefix="",
+                         max_num=-1,
+                         vis=True,
+                         **kwargs):
+    """
+    将labelme格式转换为labelme格式
+    :param data_root:  输入labelme根目录
+    :param out_root:  输出labelme根目录
+    :param class_name: 需要选择的类别，None表示全部
+    :param class_dict: 类别映射
+    :param prefix: 提供文件名前缀，则进行重新命名
+    :param max_num: 最多转换样本个数
+    """
+    if data_root and not out_root: out_root = os.path.join(data_root, "labelme")
+    dataset = parser_yolo.YOLODataset(filename=None,
+                                      data_root=data_root,
+                                      anno_dir=None,
+                                      image_dir=None,
+                                      class_name=class_name,
+                                      check=False,
+                                      phase="val",
+                                      shuffle=False)
+    print("have num:{}".format(len(dataset)))
+    nums = min(len(dataset), max_num) if max_num > 0 else len(dataset)
+    for i in tqdm(range(nums)):
+        data_info = dataset.__getitem__(i)
+        image, names, points = data_info["image"], data_info["names"], data_info["points"]
+        image_file = data_info["image_file"]
+        build_labelme.save_labelme(out_root, image_file=image_file, points=points, names=names,
+                                   class_dict=class_dict, image=image, prefix=prefix, index=i, vis=vis)
+    file_utils.write_data(os.path.join(out_root, "class_name.txt"), dataset.class_name)
+
+
+if __name__ == "__main__":
+    class_name = ['AngelFish', 'BlueTang', 'ButterflyFish', 'ClownFish', 'GoldFish', 'Gourami', 'MorishIdol',
+                  'PlatyFish', 'RibbonedSweetlips', 'ThreeStripedDamselfish', 'YellowCichlid', 'YellowTang',
+                  'ZebraFish']
+    data_root = "/home/PKing/nasdata/tmp/tmp/Fish/test/labelme/images"
+    convert_yolo2labelme(data_root, class_name=class_name, prefix="test", vis=False)

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/converter/convert_yolo2voc.py

@@ -1,10 +1,13 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2022-10-12 16:21:55
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
+
 import os
 import numpy as np
 import cv2

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/cvutils/video_utils.py

@@ -212,7 +212,7 @@ def resize_video(video_file, save_video, size=(), start=0, interval=1, vis=True,
     video_writer.release()
 
 
-def video_capture(video_file: int or str, save_video: str or int = None, interval=1, task: Callable = None,
+def video_capture(video_file: int or str, save_video: str or int = None, interval=1, freq=0, task: Callable = None,
                   vis=True, **kwargs):
     """
     读取摄像头或者视频流
@@ -220,6 +220,7 @@ def video_capture(video_file: int or str, save_video: str or int = None, interva
                        Int 摄像头ID，如0，1，2
     :param save_video: 保存task视频处理后的结果
     :param interval: 抽帧处理间隔
+    :param freq: 抽帧频率，当freq>0，表示interval=int(fps / freq)
     :param task: 回调函数： def task(frame, **kwargs)
     :param kwargs:回调函数输入参数,
                  delay: 控制显示延时
@@ -234,6 +235,7 @@ def video_capture(video_file: int or str, save_video: str or int = None, interva
     if num_frames <= 0: num_frames = end
     end = min(end, num_frames)  # TODO 当num_frames<0时，使用0<end<count继续播放
     interval = fps if interval == -1 else interval  # 当interval=-1，表示interval=fps,即一秒一帧
+    interval = int(fps / freq) if freq > 0 else interval
     save_fps = max(kwargs.get("speed", 1) * fps // interval, 1)
     count = 0
     video_writer = None
@@ -257,7 +259,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 or str, save_video: str or int = None, interval=1, task: Callable = None,
+def video_iterator(video_file: int or str, save_video: str or int = None, interval=1, freq=0, task: Callable = None,
                    vis=False, **kwargs):
     """
     读取摄像头或者视频流迭代器
@@ -271,6 +273,7 @@ def video_iterator(video_file: int or str, save_video: str or int = None, interv
                        Int 摄像头ID，如0，1，2
     :param save_video: 保存task视频处理后的结果
     :param interval: 抽帧处理间隔，当interval=-1，表示当interval=fps,即一秒一帧
+    :param freq: 抽帧频率，当freq>0，表示interval=int(fps / freq)
     :param task: 回调函数： def task(frame, **kwargs)
     :param kwargs:回调函数输入参数,
                  delay: 控制显示延时,默认10S
@@ -287,6 +290,7 @@ def video_iterator(video_file: int or str, save_video: str or int = None, interv
     end = int(kwargs.get("end", num_frames / fps) * fps) if fps > 0 else 0
     end = min(end, num_frames) if end > 0 else num_frames  # TODO 当num_frames<0时，使用0<end<count继续播放
     interval = fps if interval == -1 else interval  # 当interval=-1，表示interval=fps,即一秒一帧
+    interval = int(fps / freq) if freq > 0 else interval
     save_fps = max(kwargs.get("speed", 1) * fps // interval, 1)
     count = 0
     video_writer = None

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/dataloader/base_coco.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : PKing
-    @E-mail : 
-    @Date   : 2023-08-10 10:18:32
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np
@@ -365,8 +367,8 @@ class CocoDataset(object):
                 continue
             keypoint = ann.get('keypoints', [])
             if len(keypoint) == 0: continue
-            keypoint = np.asarray(keypoint).reshape(num_joints, 3)
-            keypoint = keypoint[:, 0:2]
+            keypoint = np.asarray(keypoint).reshape(num_joints, 3)  # (N,3)
+            # keypoint = keypoint[:, 0:2]
             keypoints.append(keypoint)
             label = self.class_dict[name]
             rects.append(ann['bbox'])

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/dataloader/base_dataset.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : PKing
-    @E-mail :
-    @Date   : 2023-08-10 10:18:32
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/dataloader/data_resample.py

@@ -1,8 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail :
-    @Date   : 2021-04-21 09:41:34
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import random
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/dataloader/parser_coco_det.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : PKing
-    @E-mail :
-    @Date   : 2023-08-10 10:18:32
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/dataloader/parser_coco_ins.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : PKing
-    @E-mail :
-    @Date   : 2023-08-10 10:18:32
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np

{pybaseutils-2.3.2 → pybaseutils-2.3.4}/pybaseutils/dataloader/parser_coco_kps.py

@@ -1,9 +1,11 @@
 # -*- coding: utf-8 -*-
 """
-    @Author : PKing
-    @E-mail :
-    @Date   : 2023-08-10 10:18:32
-    @Brief  :
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-04-29 09:13:09
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
 import numpy as np
@@ -71,6 +73,7 @@ class CocoKeypoint(base_coco.CocoDataset):
         anns_info, file_info = self.get_object_annotations(image_id)
         image, width, height, image_file = self.get_object_image(file_info)
         boxes, labels, keypoints = self.get_keypoint_info(anns_info, self.num_joints)
+        # TODO keypoints is (1,N,3); v=0未标注点; v=1标注了但是图像中不可见（例如遮挡）;v=2标注了并图像可见
         data = {"image": image, "boxes": boxes, "labels": labels,
                 "segs": [], "mask": [], "keypoints": keypoints, "target": [],
                 "image_id": image_id, "annotations": anns_info, "file_info": file_info,
@@ -135,16 +138,16 @@ if __name__ == "__main__":
     # image_dir = coco_root + 'val2017/images'
     # anno_file = coco_root + 'annotations/person_keypoints_val2017.json'
 
-    image_dir = "/media/PKing/新加卷1/SDK/base-utils/data/coco/JPEGImages"
-    anno_file = "/media/PKing/新加卷1/SDK/base-utils/data/coco/coco_kps.json"
+    image_dir = "../../data/coco/JPEGImages"
+    anno_file = "../../data/coco/coco_kps.json"
     class_name = ["person"]
 
     # hand
-    image_dir = ""
+    # image_dir = ""
     # anno_file = "/home/PKing/nasdata/dataset/tmp/hand-pose/HandPose-v2/train/train_anno.json"
     # anno_file = "/home/PKing/nasdata/dataset/tmp/hand-pose/HandPose-v1/test/test_anno.json"
-    anno_file = "/home/PKing/nasdata/dataset/tmp/hand-pose/HandPose-v2/train/train_anno.json"
-    class_name = []
+    # anno_file = "/home/PKing/nasdata/dataset/tmp/hand-pose/HandPose-v2/train/train_anno.json"
+    # class_name = []
     dataset = CocoKeypoint(anno_file, image_dir, class_name=class_name)
     skeleton = dataset.skeleton
     for i in range(len(dataset)):