pms-nvidia-processor 2.2.0__tar.gz → 2.2.1__tar.gz

{pms_nvidia_processor-2.2.0 → pms_nvidia_processor-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pms-nvidia-processor
-Version: 2.2.0
+Version: 2.2.1
 Summary: lib for pms nvidia processors
 Author: HyeongSeok Kim
 Author-email: tiryul@gmail.com

{pms_nvidia_processor-2.2.0 → pms_nvidia_processor-2.2.1}/pms_nvidia_processor/__init__.py

@@ -10,4 +10,4 @@ from .processor.color_resnet.processor_pre_v2 import COLORRESNETPREProcessor
 from .processor.color_resnet.processor_post_v2 import COLORRESNETPOSTProcessor
 from .processor.gg.processor_v2 import GGProcessor
 
-__version__ = "2.2.0"
+__version__ = "2.2.1"

{pms_nvidia_processor-2.2.0 → pms_nvidia_processor-2.2.1}/pms_nvidia_processor/utility/caster.py

@@ -2,6 +2,9 @@ import numpy as np
 
 
 class Caster:
+    __auto_clip_map = {
+        np.dtype("uint8"): (0, 255),
+    }
 
     def __init__(self, input_vector: np.ndarray) -> None:
         self._castable_type = input_vector.dtype
@@ -17,8 +20,12 @@ class Caster:
         assert (
             vector.dtype == self._uncastable_type
         ), f"ERROR, The vector's type must be {self._uncastable_type}"
-        return np.clip(
-            vector,
-            np.iinfo(self._castable_type).min,
-            np.iinfo(self._castable_type).max,
-        ).astype(self._castable_type)
+        if self._castable_type in self.__auto_clip_map:
+            min, max = self.__auto_clip_map[self._castable_type]
+            return np.clip(
+                vector,
+                min,
+                max,
+            ).astype(self._castable_type)
+        else:
+            return vector.astype(self._castable_type)

{pms_nvidia_processor-2.2.0 → pms_nvidia_processor-2.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pms-nvidia-processor"
-version = "2.2.0"
+version = "2.2.1"
 description = "lib for pms nvidia processors"
 authors = ["HyeongSeok Kim <tiryul@gmail.com>"]
 readme = "README.md"