pyanime4k 3.1.0__cp310-cp310-win_amd64.whl

pyanime4k/__init__.py

@@ -0,0 +1,3 @@
+from .info import *
+from .upscale import *
+from .pyac import *

pyanime4k/info.py

@@ -0,0 +1,13 @@
+from .pyac import pyac
+
+def print_model_list():
+    for name, description in zip(pyac.specs.ModelNameList, pyac.specs.ModelDescriptionList):
+        print(f"{name}\t\t{description}")
+
+def print_processor_list():
+    for name, description in zip(pyac.specs.ProcessorNameList, pyac.specs.ProcessorDescriptionList):
+        print(f"{name}\t\t{description}")
+
+def print_device_list():
+    for info in pyac.core.Processor.InfoList:
+        print(info, end="")

pyanime4k/pyac/__init__.py

@@ -0,0 +1 @@
+from .pyac.core import *

pyanime4k/pyac/pyac/__init__.pyi

@@ -0,0 +1,7 @@
+"""
+Anime4KCPP: A high performance anime upscaler.
+"""
+from __future__ import annotations
+from . import core
+from . import specs
+__all__: list[str] = ['core', 'specs']

pyanime4k/pyac/pyac/core.pyi

@@ -0,0 +1,184 @@
+from __future__ import annotations
+import numpy
+import numpy.typing
+import typing
+__all__: list[str] = ['IMREAD_COLOR', 'IMREAD_GRAYSCALE', 'IMREAD_RGB', 'IMREAD_RGBA', 'IMREAD_UNCHANGED', 'ImreadModes', 'Processor', 'RESIZE_BICUBIC_0_100', 'RESIZE_BICUBIC_0_60', 'RESIZE_BICUBIC_0_75', 'RESIZE_BICUBIC_20_50', 'RESIZE_BILINEAR', 'RESIZE_CATMULL_ROM', 'RESIZE_LANCZOS2', 'RESIZE_LANCZOS3', 'RESIZE_LANCZOS4', 'RESIZE_MITCHELL_NETRAVALI', 'RESIZE_POINT', 'RESIZE_SOFTCUBIC100', 'RESIZE_SOFTCUBIC50', 'RESIZE_SOFTCUBIC75', 'RESIZE_SPLINE16', 'RESIZE_SPLINE36', 'RESIZE_SPLINE64', 'ResizeModes', 'imread', 'imwrite', 'resize']
+class ImreadModes:
+    """
+    Members:
+    
+      IMREAD_UNCHANGED
+    
+      IMREAD_GRAYSCALE
+    
+      IMREAD_COLOR
+    
+      IMREAD_RGB
+    
+      IMREAD_RGBA
+    """
+    IMREAD_COLOR: typing.ClassVar[ImreadModes]  # value = <ImreadModes.IMREAD_COLOR: 3>
+    IMREAD_GRAYSCALE: typing.ClassVar[ImreadModes]  # value = <ImreadModes.IMREAD_GRAYSCALE: 1>
+    IMREAD_RGB: typing.ClassVar[ImreadModes]  # value = <ImreadModes.IMREAD_COLOR: 3>
+    IMREAD_RGBA: typing.ClassVar[ImreadModes]  # value = <ImreadModes.IMREAD_RGBA: 4>
+    IMREAD_UNCHANGED: typing.ClassVar[ImreadModes]  # value = <ImreadModes.IMREAD_UNCHANGED: 0>
+    __members__: typing.ClassVar[dict[str, ImreadModes]]  # value = {'IMREAD_UNCHANGED': <ImreadModes.IMREAD_UNCHANGED: 0>, 'IMREAD_GRAYSCALE': <ImreadModes.IMREAD_GRAYSCALE: 1>, 'IMREAD_COLOR': <ImreadModes.IMREAD_COLOR: 3>, 'IMREAD_RGB': <ImreadModes.IMREAD_COLOR: 3>, 'IMREAD_RGBA': <ImreadModes.IMREAD_RGBA: 4>}
+    def __eq__(self, other: typing.Any) -> bool:
+        ...
+    def __getstate__(self) -> int:
+        ...
+    def __hash__(self) -> int:
+        ...
+    def __index__(self) -> int:
+        ...
+    def __init__(self, value: typing.SupportsInt) -> None:
+        ...
+    def __int__(self) -> int:
+        ...
+    def __ne__(self, other: typing.Any) -> bool:
+        ...
+    def __repr__(self) -> str:
+        ...
+    def __setstate__(self, state: typing.SupportsInt) -> None:
+        ...
+    def __str__(self) -> str:
+        ...
+    @property
+    def name(self) -> str:
+        ...
+    @property
+    def value(self) -> int:
+        ...
+class Processor:
+    CPU: typing.ClassVar[int] = 0
+    CUDA: typing.ClassVar[int] = 2
+    OpenCL: typing.ClassVar[int] = 1
+    InfoList: typing.ClassVar[tuple[str, ...]]
+    def __call__(self, src: numpy.ndarray, factor: typing.SupportsFloat = 2.0) -> numpy.ndarray:
+        ...
+    @typing.overload
+    def __init__(self, processor_type: typing.SupportsInt = 0, device: typing.SupportsInt = 0, model: str = 'acnet-gan') -> None:
+        ...
+    @typing.overload
+    def __init__(self, processor_type: str, device: typing.SupportsInt = 0, model: str = 'acnet-gan') -> None:
+        ...
+    def __str__(self) -> str:
+        ...
+    def error(self) -> str:
+        ...
+    def name(self) -> str:
+        ...
+    def ok(self) -> bool:
+        ...
+    def process(self, src: numpy.ndarray, factor: typing.SupportsFloat = 2.0) -> numpy.ndarray:
+        ...
+class ResizeModes:
+    """
+    Members:
+    
+      RESIZE_POINT
+    
+      RESIZE_CATMULL_ROM
+    
+      RESIZE_MITCHELL_NETRAVALI
+    
+      RESIZE_BICUBIC_0_60
+    
+      RESIZE_BICUBIC_0_75
+    
+      RESIZE_BICUBIC_0_100
+    
+      RESIZE_BICUBIC_20_50
+    
+      RESIZE_SOFTCUBIC50
+    
+      RESIZE_SOFTCUBIC75
+    
+      RESIZE_SOFTCUBIC100
+    
+      RESIZE_LANCZOS2
+    
+      RESIZE_LANCZOS3
+    
+      RESIZE_LANCZOS4
+    
+      RESIZE_SPLINE16
+    
+      RESIZE_SPLINE36
+    
+      RESIZE_SPLINE64
+    
+      RESIZE_BILINEAR
+    """
+    RESIZE_BICUBIC_0_100: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_BICUBIC_0_100: 5>
+    RESIZE_BICUBIC_0_60: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_BICUBIC_0_60: 3>
+    RESIZE_BICUBIC_0_75: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_BICUBIC_0_75: 4>
+    RESIZE_BICUBIC_20_50: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_BICUBIC_20_50: 6>
+    RESIZE_BILINEAR: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_BILINEAR: 16>
+    RESIZE_CATMULL_ROM: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_CATMULL_ROM: 1>
+    RESIZE_LANCZOS2: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_LANCZOS2: 10>
+    RESIZE_LANCZOS3: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_LANCZOS3: 11>
+    RESIZE_LANCZOS4: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_LANCZOS4: 12>
+    RESIZE_MITCHELL_NETRAVALI: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_MITCHELL_NETRAVALI: 2>
+    RESIZE_POINT: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_POINT: 0>
+    RESIZE_SOFTCUBIC100: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_SOFTCUBIC100: 9>
+    RESIZE_SOFTCUBIC50: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_SOFTCUBIC50: 7>
+    RESIZE_SOFTCUBIC75: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_SOFTCUBIC75: 8>
+    RESIZE_SPLINE16: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_SPLINE16: 13>
+    RESIZE_SPLINE36: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_SPLINE36: 14>
+    RESIZE_SPLINE64: typing.ClassVar[ResizeModes]  # value = <ResizeModes.RESIZE_SPLINE64: 15>
+    __members__: typing.ClassVar[dict[str, ResizeModes]]  # value = {'RESIZE_POINT': <ResizeModes.RESIZE_POINT: 0>, 'RESIZE_CATMULL_ROM': <ResizeModes.RESIZE_CATMULL_ROM: 1>, 'RESIZE_MITCHELL_NETRAVALI': <ResizeModes.RESIZE_MITCHELL_NETRAVALI: 2>, 'RESIZE_BICUBIC_0_60': <ResizeModes.RESIZE_BICUBIC_0_60: 3>, 'RESIZE_BICUBIC_0_75': <ResizeModes.RESIZE_BICUBIC_0_75: 4>, 'RESIZE_BICUBIC_0_100': <ResizeModes.RESIZE_BICUBIC_0_100: 5>, 'RESIZE_BICUBIC_20_50': <ResizeModes.RESIZE_BICUBIC_20_50: 6>, 'RESIZE_SOFTCUBIC50': <ResizeModes.RESIZE_SOFTCUBIC50: 7>, 'RESIZE_SOFTCUBIC75': <ResizeModes.RESIZE_SOFTCUBIC75: 8>, 'RESIZE_SOFTCUBIC100': <ResizeModes.RESIZE_SOFTCUBIC100: 9>, 'RESIZE_LANCZOS2': <ResizeModes.RESIZE_LANCZOS2: 10>, 'RESIZE_LANCZOS3': <ResizeModes.RESIZE_LANCZOS3: 11>, 'RESIZE_LANCZOS4': <ResizeModes.RESIZE_LANCZOS4: 12>, 'RESIZE_SPLINE16': <ResizeModes.RESIZE_SPLINE16: 13>, 'RESIZE_SPLINE36': <ResizeModes.RESIZE_SPLINE36: 14>, 'RESIZE_SPLINE64': <ResizeModes.RESIZE_SPLINE64: 15>, 'RESIZE_BILINEAR': <ResizeModes.RESIZE_BILINEAR: 16>}
+    def __eq__(self, other: typing.Any) -> bool:
+        ...
+    def __getstate__(self) -> int:
+        ...
+    def __hash__(self) -> int:
+        ...
+    def __index__(self) -> int:
+        ...
+    def __init__(self, value: typing.SupportsInt) -> None:
+        ...
+    def __int__(self) -> int:
+        ...
+    def __ne__(self, other: typing.Any) -> bool:
+        ...
+    def __repr__(self) -> str:
+        ...
+    def __setstate__(self, state: typing.SupportsInt) -> None:
+        ...
+    def __str__(self) -> str:
+        ...
+    @property
+    def name(self) -> str:
+        ...
+    @property
+    def value(self) -> int:
+        ...
+def imread(filename: str, mode: ImreadModes = ...) -> numpy.typing.NDArray[numpy.uint8]:
+    ...
+def imwrite(filename: str, image: typing.Annotated[numpy.typing.ArrayLike, numpy.uint8]) -> bool:
+    ...
+def resize(src: numpy.ndarray, dsize: tuple, fx: typing.SupportsFloat = 0.0, fy: typing.SupportsFloat = 0.0, mode: ResizeModes = ...) -> numpy.ndarray:
+    ...
+IMREAD_COLOR: ImreadModes  # value = <ImreadModes.IMREAD_COLOR: 3>
+IMREAD_GRAYSCALE: ImreadModes  # value = <ImreadModes.IMREAD_GRAYSCALE: 1>
+IMREAD_RGB: ImreadModes  # value = <ImreadModes.IMREAD_COLOR: 3>
+IMREAD_RGBA: ImreadModes  # value = <ImreadModes.IMREAD_RGBA: 4>
+IMREAD_UNCHANGED: ImreadModes  # value = <ImreadModes.IMREAD_UNCHANGED: 0>
+RESIZE_BICUBIC_0_100: ResizeModes  # value = <ResizeModes.RESIZE_BICUBIC_0_100: 5>
+RESIZE_BICUBIC_0_60: ResizeModes  # value = <ResizeModes.RESIZE_BICUBIC_0_60: 3>
+RESIZE_BICUBIC_0_75: ResizeModes  # value = <ResizeModes.RESIZE_BICUBIC_0_75: 4>
+RESIZE_BICUBIC_20_50: ResizeModes  # value = <ResizeModes.RESIZE_BICUBIC_20_50: 6>
+RESIZE_BILINEAR: ResizeModes  # value = <ResizeModes.RESIZE_BILINEAR: 16>
+RESIZE_CATMULL_ROM: ResizeModes  # value = <ResizeModes.RESIZE_CATMULL_ROM: 1>
+RESIZE_LANCZOS2: ResizeModes  # value = <ResizeModes.RESIZE_LANCZOS2: 10>
+RESIZE_LANCZOS3: ResizeModes  # value = <ResizeModes.RESIZE_LANCZOS3: 11>
+RESIZE_LANCZOS4: ResizeModes  # value = <ResizeModes.RESIZE_LANCZOS4: 12>
+RESIZE_MITCHELL_NETRAVALI: ResizeModes  # value = <ResizeModes.RESIZE_MITCHELL_NETRAVALI: 2>
+RESIZE_POINT: ResizeModes  # value = <ResizeModes.RESIZE_POINT: 0>
+RESIZE_SOFTCUBIC100: ResizeModes  # value = <ResizeModes.RESIZE_SOFTCUBIC100: 9>
+RESIZE_SOFTCUBIC50: ResizeModes  # value = <ResizeModes.RESIZE_SOFTCUBIC50: 7>
+RESIZE_SOFTCUBIC75: ResizeModes  # value = <ResizeModes.RESIZE_SOFTCUBIC75: 8>
+RESIZE_SPLINE16: ResizeModes  # value = <ResizeModes.RESIZE_SPLINE16: 13>
+RESIZE_SPLINE36: ResizeModes  # value = <ResizeModes.RESIZE_SPLINE36: 14>
+RESIZE_SPLINE64: ResizeModes  # value = <ResizeModes.RESIZE_SPLINE64: 15>

pyanime4k/pyac/pyac/specs.pyi

@@ -0,0 +1,6 @@
+from __future__ import annotations
+__all__: list[str] = ['ModelDescriptionList', 'ModelNameList', 'ProcessorDescriptionList', 'ProcessorNameList']
+ModelDescriptionList: tuple = ('Lightweight CNN, detail enhancement.', 'Lightweight CNN, mild denoising.', 'Lightweight CNN, moderate denoising.', 'Lightweight CNN, heavy denoising.', 'Lightweight CNN, extreme denoising.', 'Lightweight ResNet, mild denoising.')
+ModelNameList: tuple = ('acnet-gan', 'acnet-hdn0', 'acnet-hdn1', 'acnet-hdn2', 'acnet-hdn3', 'arnet-hdn')
+ProcessorDescriptionList: tuple = ('General-purpose CPU processing with optional SIMD acceleration.', 'Cross-platform acceleration requiring OpenCL 1.2+ compliant devices.', 'NVIDIA GPU acceleration requiring Compute Capability 5.0+.')
+ProcessorNameList: tuple = ('cpu', 'opencl', 'cuda')

pyanime4k/upscale.py

@@ -0,0 +1,28 @@
+from itertools import zip_longest, islice
+from pathlib import Path
+
+from .pyac import pyac
+
+def upscale_images(
+    inputs: list,
+    outputs: list = [],
+    output_suffix: str = "_output",
+    factor: float = 2.0,
+    processor_type: str = "cpu",
+    device_id:int = 0,
+    model_name: str = "acnet-hdn0"
+):
+    if isinstance(inputs, str):
+        inputs = [inputs]
+
+    processor = pyac.core.Processor(processor_type, device_id, model_name)
+
+    for input, output in islice(zip_longest(inputs, outputs, fillvalue=None), len(inputs)):
+        if output is None:
+            input_path = Path(input)
+            output = input_path.with_stem(input_path.stem + output_suffix)
+
+        src = pyac.core.imread(str(input))
+        dst = processor(src, factor)
+        if not pyac.core.imwrite(str(output), dst):
+            raise IOError(f"Failed to save image to {output}")

pyanime4k-3.1.0.dist-info/METADATA

@@ -0,0 +1,122 @@
+Metadata-Version: 2.4
+Name: pyanime4k
+Version: 3.1.0
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: MacOS
+Classifier: Operating System :: POSIX
+Classifier: Operating System :: Unix
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Dynamic: license-file
+
+# PyAnime4K
+
+PyAnime4K is a high performance anime image upscaler based on [Anime4KCPP](https://github.com/TianZerL/Anime4KCPP).
+
+# Install
+PyAnime4K can be installed easily through pip.
+```shell
+pip install pyanime4k
+```
+
+You can also build the wheel by yourself.
+```shell
+# sudo apt install ocl-icd-opencl-dev cmake build-essential
+get clone --recurse-submodules https://github.com/TianZerL/pyanime4k.git
+cd pyanime4k
+pip install scikit-build
+python setup.py bdist_wheel
+```
+
+# Usages
+## Printing Info
+```python
+import pyanime4k
+
+# Print supported model
+pyanime4k.print_model_list()
+
+# Print supported processor
+pyanime4k.print_processor_list()
+
+# Print supported devices for image processing
+pyanime4k.print_device_list()
+```
+## Upscaling Images
+```python
+import pyanime4k
+
+# with OpenCL acceleration, if possible
+processor_type = "opencl"
+
+# upscale a single image
+pyanime4k.upscale_images("image1.png", processor_type = processor_type)
+
+# upscale a list of images
+pyanime4k.upscale_images(["image1.png", "image2.png"], processor_type = processor_type)
+```
+## Manual Upscaling
+```python
+import pyanime4k
+
+# Create a processor
+processor = pyanime4k.Processor(
+    processor_type="cpu",
+    device=0,
+    model="acnet-hdn0"
+)
+# Print processor info
+print(processor)
+# Read an image
+src = pyanime4k.imread("image1.png")
+# Process it
+dst = processor(src=src, factor=2.0)
+# Write to disk
+pyanime4k.imwrite("image1_outout.png", dst)
+```
+## OpenCV
+``` Python
+import cv2, pyanime4k
+
+img = cv2.imread("image.png")
+
+processor = pyanime4k.Processor(
+    processor_type="cpu",
+    device=0,
+    model="acnet-hdn0"
+)
+
+# opencv load image as BGR, but we need an RGB image
+src = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+
+dst = processor(src)
+
+img = cv2.cvtColor(dst, cv2.COLOR_RGB2BGR)
+
+cv2.imshow("pyanime4k", img)
+cv2.waitKey()
+```
+## Pillow
+```python
+from PIL import Image
+import numpy, pyanime4k
+
+img = Image.open("D:/temp/p1.png")
+
+processor = pyanime4k.Processor(
+    processor_type="cpu",
+    device=0,
+    model="acnet-hdn0"
+)
+
+# We need a numpy array
+src = numpy.asarray(img)
+dst = processor(src)
+img = Image.fromarray(dst)
+
+img.show()
+```

pyanime4k-3.1.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+pyanime4k/__init__.py,sha256=OYVRiJgxr163XyjVRh1x5ZNX1UOZHBv9GDBOp9buj64,66
+pyanime4k/info.py,sha256=ZgSahqzqxSR_lrXQAi3rOZhO4lLorasp5NfaqV5uljY,468
+pyanime4k/upscale.py,sha256=GoH2oh8fYX5TdIfDCBEB-xxsf_3xAH_ARam11hzrxUA,899
+pyanime4k/pyac/__init__.py,sha256=OoffRsy0diunqy8QzvD9WhSxcGhbfzD499PghpbLRs0,26
+pyanime4k/pyac/pyac.cp310-win_amd64.pyd,sha256=UnLLYF2KD4M9kqyuhSMe7yrYpCnRjGGfyVkGO4c7DFI,2248192
+pyanime4k/pyac/pyac/__init__.pyi,sha256=CBaI96ey60K1GjfSjj4VksuUpc7NbtcYFcRiGU5SJuk,175
+pyanime4k/pyac/pyac/core.pyi,sha256=SKu1Pt4wFyoQZ1TeEvNNEE_FI8pTL6bwcw6HO3Sel6Y,9187
+pyanime4k/pyac/pyac/specs.pyi,sha256=YoNQA6yXEglvAsaJixnXVIB4irLusAqnRfMUaoHLQ9c,810
+pyanime4k-3.1.0.dist-info/licenses/LICENSE,sha256=Y1LqP77JBqrXWO4jJ_2bg6N-hHoU_JGgxQnYqxw1f3U,1085
+pyanime4k-3.1.0.dist-info/METADATA,sha256=rr_b8UkJn-z0jpmN6vAffKLYqlGhOYQqWPkKp_wCwF8,2665
+pyanime4k-3.1.0.dist-info/WHEEL,sha256=grXnj_hjz9AcJWnDGoMAyy0X5GgALMAtIvNbRMBzgmk,96
+pyanime4k-3.1.0.dist-info/top_level.txt,sha256=KxVm1dNU5ew2loJ2VSTl_3lkuv8_VWFpd0xSjcTtLqA,21
+pyanime4k-3.1.0.dist-info/RECORD,,

pyanime4k-3.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: skbuild 0.18.1
+Root-Is-Purelib: false
+Tag: cp310-cp310-win_amd64
+

pyanime4k-3.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 TianZer
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

pyanime4k-3.1.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+Anime4KCPP
+pyanime4k