pixelart.py

"""
@author: Manny Gonzalez
@title: 🐯 YFG Comical Nodes
@nickname: 🐯 YFG Comical Nodes
@description: Utility custom nodes for special effects, image manipulation and quality of life tools.
"""

## Based on original code by XSS https://civitai.com/models/24869?modelVersionId=29750 ##

import torch

class PixelArt:
    interpolation_methods = ['nearest', 'bilinear', 'bicubic', 'area', 'nearest-exact']

    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "image": ("IMAGE",),
                "pixel_size": ("INT", {
                    "default": 4, 
                    "min": 1,
                    "max": 512,
                    "step": 1
                }),
                "interpolation_method": (s.interpolation_methods, {"default": s.interpolation_methods[0]}),
            },
        }

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "mosaic"

    CATEGORY = "🐯 YFG"

    def mosaic(self, image, interpolation_method, pixel_size):
        samples = image.movedim(-1,1)

        starting_width = samples.shape[3]
        starting_height = samples.shape[2]

        #downsample dimensions
        dowsample_width = int(starting_width / pixel_size) | 1
        dowsample_height = int(starting_height / pixel_size) | 1

        downsampled_image = torch.nn.functional.interpolate(samples, size=(dowsample_height, dowsample_width), mode=interpolation_method)
        output_image = torch.nn.functional.interpolate(downsampled_image, size=(starting_height, starting_width), mode='nearest')

        output = output_image.movedim(1,-1)

        return (output,)