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.
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/PKG-INFO +1 -1
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/__init__.py +3 -3
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_color_resnet_post_processor.py +1 -1
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_color_resnet_pre_processor.py +1 -1
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_const.py +1 -1
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pyproject.toml +1 -1
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/README.md +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dependency.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dpir_processor.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_asm_sr_f3_processor.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_rbpn_deinter_f3_gloss_processor.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_rbpn_sr_f3_processor.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dru_rbpn_sr_f5_processor.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_fisf_processor.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_nvidia_processor_base.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/__init__.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patch_collection.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patch_position.py +0 -0
- {pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patcher.py +0 -0
|
@@ -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
|
|
8
|
-
from ._color_resnet_post_processor import
|
|
7
|
+
from ._color_resnet_pre_processor import COLORRESNETPREProcessor
|
|
8
|
+
from ._color_resnet_post_processor import COLORRESNETPOSTProcessor
|
|
9
9
|
|
|
10
|
-
__version__ = "1.3.
|
|
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
|
|
73
|
+
class COLORRESNETPOSTProcessor(NVIDIAProcessorBase[EngineIOData, EngineIOData]):
|
|
74
74
|
|
|
75
75
|
def __init__(
|
|
76
76
|
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 =
|
|
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
|
|
File without changes
|
{pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dependency.py
RENAMED
|
File without changes
|
{pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_dpir_processor.py
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/_fisf_processor.py
RENAMED
|
File without changes
|
|
File without changes
|
{pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/__init__.py
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|
{pms_nvidia_processor-1.3.0 → pms_nvidia_processor-1.3.2}/pms_nvidia_processor/patcher/_patcher.py
RENAMED
|
File without changes
|