PyPI - pybaseutils - Versions diffs - 2.2.17__tar.gz → 2.2.19__tar.gz - Mend

pybaseutils 2.2.17tar.gz → 2.2.19tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (280) hide show

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pybaseutils
-Version: 2.2.17
+Version: 2.2.19
 Summary: pybaseutils
 Home-page: https://github.com/PanJinquan/base-utils
 Author: PanJinquan

pybaseutils-2.2.19/pybaseutils/__init__.py ADDED Viewed

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-02-18 16:10:49
+# @Brief  :
+# --------------------------------------------------------
+"""
+__version__ = '2.2.19'

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/audio/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author :
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/audio/audio_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author :
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/audio/pyaudio_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/audio/vad_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/base64_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/batch_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/build_utils/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/build_utils/cython_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/build_utils/pyarmor_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17/pybaseutils/converter → pybaseutils-2.2.19/pybaseutils/cluster}/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cluster/kmean.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cluster/maxmin_distance.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Project: IntelligentManufacture
     @File   : maxmin_distance.py

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cluster/similarity.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/config_utils.py RENAMED Viewed

@@ -1,4 +1,12 @@
-# -*- coding:utf-8 -*-
+# -*- coding: utf-8 -*-
+"""
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-02-18 16:10:49
+# @Brief  :
+# --------------------------------------------------------
+"""
 import os
 import argparse
 import numbers

{pybaseutils-2.2.17/pybaseutils/cluster → pybaseutils-2.2.19/pybaseutils/converter}/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/build_cvat.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/build_labelme.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/build_voc.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_coco2labelme.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_coco2voc.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_cvat2labelme.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_labelme2cvat.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_voc2labelme.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_voc2voc.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_voc2yolo.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/converter/convert_yolo2voc.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/coords_utils.py RENAMED Viewed

@@ -1,8 +1,11 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2021-09-08 16:58:52
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-02-18 16:10:49
+# @Brief  :
+# --------------------------------------------------------
 """
 import math

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cvutils/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cvutils/corner_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cvutils/mouse_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cvutils/nms_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/cvutils/video_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com
@@ -242,6 +242,7 @@ def video_capture(video_file: int or str, save_video: str or int = None, interva
     w, h, num_frames, fps = image_utils.get_video_info(video_cap)
     start = int(kwargs.get("start", 0) * fps)
     end = int(kwargs.get("end", num_frames / fps) * fps) if fps > 0 else 0
+    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,即一秒一帧
     save_fps = max(kwargs.get("speed", 1) * fps // interval, 1)

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/balanced_classes.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : panjq
     @E-mail : pan_jinquan@163.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/base_coco.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/base_dataset.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :
@@ -67,8 +67,6 @@ class Dataset(object):
         """
         if isinstance(class_name, str):
             class_name = Dataset.read_file(class_name)
-        elif isinstance(class_name, list) and "unique" in class_name:
-            self.unique = True
         if isinstance(class_name, list) and len(class_name) > 0:
             class_dict = {}
             for i, name in enumerate(class_name):
@@ -85,6 +83,8 @@ class Dataset(object):
             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())
+        if isinstance(class_name, list) and "unique" in class_name:
+            self.unique = True
         return class_name, class_dict
     @staticmethod

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/data_resample.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/parser_coco_det.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/parser_coco_ins.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/parser_coco_kps.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/parser_labelme.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :
@@ -193,6 +193,7 @@ class LabelMeDataset(Dataset):
         bboxes, labels, points, groups, names, keypoints = [], [], [], [], [], []
         for anno in annotation:
             name = "unique" if unique else anno["label"]
+            shape_type = anno.get("shape_type", "polygon")
             label = name
             if class_dict:
                 if not name in class_dict:
@@ -211,6 +212,8 @@ class LabelMeDataset(Dataset):
                 pts[:, 0] = np.clip(pts[:, 0], 0, w - 1)
                 pts[:, 1] = np.clip(pts[:, 1], 0, h - 1)
             box = image_utils.polygons2boxes([pts])[0]
+            if shape_type == "rectangle":
+                pts = image_utils.boxes2polygons([box])[0]
             names.append(name)
             labels.append(label)
             bboxes.append(box)

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/dataloader/parser_yolo.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : PKing
     @E-mail :

{pybaseutils-2.2.17/pybaseutils/tracking → pybaseutils-2.2.19/pybaseutils/filter}/QueueTable.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/filter/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17/pybaseutils/tracking → pybaseutils-2.2.19/pybaseutils/filter}/demo.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/filter/kalman_filter.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/filter/mean_filter.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/filter/motion_filter.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/filter/pose_filter.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/font_style/__init__.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/geometry_tools.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Project: PyKinect2-OpenCV
     @File   : geometry_tools.py

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/heatmap_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
     @Author : Pan
     @E-mail : 390737991@qq.com

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/http_utils.py RENAMED Viewed

@@ -1,4 +1,4 @@
-# -*- coding:utf-8 -*-
+# -*- coding: utf-8 -*-
 import logging
 import time
 import requests

{pybaseutils-2.2.17 → pybaseutils-2.2.19}/pybaseutils/image_utils.py RENAMED Viewed

@@ -1,8 +1,11 @@
-# -*-coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
-    @Author : Pan
-    @E-mail : 390737991@qq.com
-    @Date   : 2019-02-14 15:34:50
+# --------------------------------------------------------
+# @Author : Pan
+# @E-mail :
+# @Date   : 2025-02-18 16:10:49
+# @Brief  :
+# --------------------------------------------------------
 """
 import os
@@ -99,34 +102,6 @@ def boxes_protection(boxes, width, height):
     return boxes
-def tensor2image(batch_tensor, index=0):
-    """
-    convert tensor to image
-    :param batch_tensor:
-    :param index:
-    :return:
-    """
-    image_tensor = batch_tensor[index, :]
-    image = np.array(image_tensor, dtype=np.float32)
-    image = np.squeeze(image)
-    image = image.transpose(1, 2, 0)  # 通道由[c,h,w]->[h,w,c]
-    return image
-def get_image_tensor(image_path, image_size, transpose=False):
-    image = read_image(image_path)
-    # transform = default_transform(image_size)
-    # torch_image = transform(image).detach().numpy()
-    image = resize_image(image, size=(int(128 * image_size[0] / 112), int(128 * image_size[1] / 112)))
-    image = center_crop(image, crop_size=image_size)
-    image_tensor = image_normalize(image, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
-    if transpose:
-        image_tensor = image_tensor.transpose(2, 0, 1)  # NHWC->NCHW
-    image_tensor = image_tensor[np.newaxis, :]
-    # std = np.std(torch_image-image_tensor)
-    return image_tensor
 def image_clip(image):
     """
     :param image:
@@ -316,6 +291,55 @@ def swap_image(image):
     return image
+def image2tensor(image, size=(), mean=(), std=()):
+    """
+    transform = transforms.Compose([
+        transforms.Resize(size=(input_size[1], input_size[0])),
+        transforms.ToTensor(),
+        transforms.Normalize(mean=rgb_mean, std=rgb_std),
+    ])
+    :param image:
+    :param input_size:
+    :return:
+    """
+    if size: image = cv2.resize(image, dsize=size)
+    image = np.array(image, dtype=np.float32) / 255.0
+    if mean: image = image - np.asarray(mean, dtype=np.float32)
+    if std: image = image / np.asarray(std, dtype=np.float32)
+    tensor = image.transpose(2, 0, 1)  # HWC->CHW
+    tensor = tensor[np.newaxis, :]
+    return tensor
+def tensor2image(tensor, mean=(), std=()):
+    """
+    :param tensor: (b,c,h,w)
+    :param index:
+    :return:
+    """
+    image = np.array(tensor, dtype=np.float32)
+    image = image.transpose(0, 2, 3, 1)  # (b0,c1,h2,w3)->(b0,h2,w3,c1)
+    if std: image = image * np.asarray(std, dtype=np.float32)
+    if mean: image = image + np.asarray(mean, dtype=np.float32)
+    image = np.clip(image * 255, 0, 255)
+    image = np.array(image, dtype=np.uint8)
+    return image
+def get_image_tensor(image_path, image_size, transpose=False):
+    image = read_image(image_path)
+    # transform = default_transform(image_size)
+    # torch_image = transform(image).detach().numpy()
+    image = resize_image(image, size=(int(128 * image_size[0] / 112), int(128 * image_size[1] / 112)))
+    image = center_crop(image, crop_size=image_size)
+    image_tensor = image_normalize(image, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
+    if transpose:
+        image_tensor = image_tensor.transpose(2, 0, 1)  # NHWC->NCHW
+    image_tensor = image_tensor[np.newaxis, :]
+    # std = np.std(torch_image-image_tensor)
+    return image_tensor
 def image_normalize(image, mean=None, std=None):
     '''
     正则化，归一化
@@ -2979,7 +3003,7 @@ def get_video_info(video_cap: cv2.VideoCapture, vis=True):
     width = int(video_cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(video_cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     num_frames = int(video_cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    fps = int(video_cap.get(cv2.CAP_PROP_FPS))
+    fps = video_cap.get(cv2.CAP_PROP_FPS)
     if vis: print("read video:width:{},height:{},fps:{},num_frames:{}".format(width, height, fps, num_frames))
     return width, height, num_frames, fps