pms-nvidia-processor 1.3.2__tar.gz → 1.4.0__tar.gz

{pms_nvidia_processor-1.3.2 → pms_nvidia_processor-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pms-nvidia-processor
-Version: 1.3.2
+Version: 1.4.0
 Summary: lib for pms nvidia processors
 Author: HyeongSeok Kim
 Author-email: tiryul@gmail.com
@@ -9,6 +9,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: albumentations (>=1.4.13,<2.0.0)
 Requires-Dist: loguru
 Requires-Dist: numpy

{pms_nvidia_processor-1.3.2 → pms_nvidia_processor-1.4.0}/pms_nvidia_processor/__init__.py

@@ -7,4 +7,4 @@ from ._fisf_processor import FISFProcessor
 from ._color_resnet_pre_processor import COLORRESNETPREProcessor
 from ._color_resnet_post_processor import COLORRESNETPOSTProcessor
 
-__version__ = "1.3.2"
+__version__ = "1.4.0"

{pms_nvidia_processor-1.3.2 → pms_nvidia_processor-1.4.0}/pms_nvidia_processor/_color_resnet_post_processor.py

@@ -90,11 +90,12 @@ class COLORRESNETPOSTProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
         self.filter_coefficient = filter_coefficient
         self.config = ColorResnetPostConfig
         self.is_scale_factor_calculated = False
+        assert True
 
     async def inference(self, batch_input_data: List[np.ndarray]) -> List[np.ndarray]:
         session = self.session
         patch_size = self.config.PATCH_SIZE
-        batch_input_data = [np.zeros_like(batch_input_data[0]), *batch_input_data]
+        # batch_input_data = [np.zeros_like(batch_input_data[0]), *batch_input_data]
         batch = len(batch_input_data)
         batch_output_data: np.ndarray = np.zeros(
             (batch, patch_size, patch_size, 3), np.float32
@@ -108,10 +109,12 @@ class COLORRESNETPOSTProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
             output_buffer=self.output_buffer,
             output_image=batch_output_data,
         )
-        return [output_data for output_data in batch_output_data[1:]]  # unpack
+        # return [output_data for output_data in batch_output_data[1:]]  # unpack
+        return [output_data for output_data in batch_output_data]  # unpack
 
     async def _run(self, input_data: EngineIOData) -> EngineIOData:
-        max_batch_size = self.io_shapes["input"][0][0] - 1
+        # max_batch_size = self.io_shapes["input"][0][0] - 1
+        max_batch_size = 1
         # 여기서 patching
         input_image: np.ndarray = input_data.frame  # type: ignore
         patcher_config = self.config.PATCHER_CONFIG
@@ -129,7 +132,6 @@ class COLORRESNETPOSTProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
         for batch_items in TRT.batch(input_patches, max_batch_size):
             ops = await self.inference(batch_input_data=batch_items)
             output_patches += ops
-
         self.patcher.merge(output_vector=output_image, patches=output_patches)
         if not self.is_scale_factor_calculated:
             self.scale_min, self.scale_max = _get_scale_factor(output_image)
@@ -163,7 +165,8 @@ class COLORRESNETPOSTProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
         n_patches = len(self.patcher.slice(input_vector=padded_input_image))
 
         # set io shape
-        self.batch_size = min(n_patches, self.config.MAX_BATCH_SIZE) + 1
+        # self.batch_size = min(n_patches + 1, self.config.MAX_BATCH_SIZE - 1)
+        self.batch_size = self.config.MAX_BATCH_SIZE
         self.io_shapes = {
             "input": (
                 [self.batch_size, *trt_config.input_shape],

{pms_nvidia_processor-1.3.2 → pms_nvidia_processor-1.4.0}/pms_nvidia_processor/_const.py

@@ -244,7 +244,8 @@ class ColorResnetPostConfig:
     NUMBER_OF_OUTPUT_CHANNELS: int = 3
     UPSCALE_RATIO: int = 1
     PATCH_SIZE = 512
-    MAX_BATCH_SIZE = 8
+    # MAX_BATCH_SIZE = 8
+    MAX_BATCH_SIZE = 1
     MIN_BATCH_SIZE = 1
     OPT_BATCH_SIZE = MAX_BATCH_SIZE // 2
     INPUT_OVERLAB_LENGTH = 16 * 2

{pms_nvidia_processor-1.3.2 → pms_nvidia_processor-1.4.0}/pyproject.toml

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