pms-nvidia-processor 1.3.0__tar.gz → 1.3.2__tar.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 (19) hide show
  1. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/PKG-INFO +1 -1
  2. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/__init__.py +3 -3
  3. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_color_resnet_post_processor.py +1 -1
  4. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_color_resnet_pre_processor.py +1 -1
  5. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_const.py +1 -1
  6. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pyproject.toml +1 -1
  7. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/README.md +0 -0
  8. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dependency.py +0 -0
  9. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dpir_processor.py +0 -0
  10. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_asm_sr_f3_processor.py +0 -0
  11. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_rbpn_deinter_f3_gloss_processor.py +0 -0
  12. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_rbpn_sr_f3_processor.py +0 -0
  13. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_rbpn_sr_f5_processor.py +0 -0
  14. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_fisf_processor.py +0 -0
  15. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_nvidia_processor_base.py +0 -0
  16. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/__init__.py +0 -0
  17. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patch_collection.py +0 -0
  18. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patch_position.py +0 -0
  19. {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patcher.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: pms-nvidia-processor
3
- Version: 1.3.0
3
+ Version: 1.3.2
4
4
  Summary: lib for pms nvidia processors
5
5
  Author: HyeongSeok Kim
6
6
  Author-email: tiryul@gmail.com
@@ -4,7 +4,7 @@ from ._dru_rbpn_sr_f5_processor import DRURBPNSRF5Processor
4
4
  from ._dru_asm_sr_f3_processor import DRUASMSRF3Processor
5
5
  from ._dru_rbpn_deinter_f3_gloss_processor import DRURBPNDEINTERF3GLOSSProcessor
6
6
  from ._fisf_processor import FISFProcessor
7
- from ._color_resnet_pre_processor import ColorResnetPreProcessor
8
- from ._color_resnet_post_processor import ColorResnetPostProcessor
7
+ from ._color_resnet_pre_processor import COLORRESNETPREProcessor
8
+ from ._color_resnet_post_processor import COLORRESNETPOSTProcessor
9
9
 
10
- __version__ = "1.3.0"
10
+ __version__ = "1.3.2"
@@ -70,7 +70,7 @@ def _get_scale_factor(output_image: np.ndarray) -> Tuple[np.ndarray, np.ndarray]
70
70
 
71
71
 
72
72
  @register
73
- class ColorResnetPostProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
73
+ class COLORRESNETPOSTProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
74
74
 
75
75
  def __init__(
76
76
  self,
@@ -28,7 +28,7 @@ def _post_processing(
28
28
 
29
29
 
30
30
  @register
31
- class ColorResnetPreProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
31
+ class COLORRESNETPREProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
32
32
 
33
33
  def __init__(
34
34
  self,
@@ -244,7 +244,7 @@ class ColorResnetPostConfig:
244
244
  NUMBER_OF_OUTPUT_CHANNELS: int = 3
245
245
  UPSCALE_RATIO: int = 1
246
246
  PATCH_SIZE = 512
247
- MAX_BATCH_SIZE = 7
247
+ MAX_BATCH_SIZE = 8
248
248
  MIN_BATCH_SIZE = 1
249
249
  OPT_BATCH_SIZE = MAX_BATCH_SIZE // 2
250
250
  INPUT_OVERLAB_LENGTH = 16 * 2
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "pms-nvidia-processor"
3
- version = "1.3.0"
3
+ version = "1.3.2"
4
4
  description = "lib for pms nvidia processors"
5
5
  authors = ["HyeongSeok Kim <tiryul@gmail.com>"]
6
6
  readme = "README.md"