JANK

A4P7J1N7M05OT · Apr 5, 2024 · 3c090de · 3c090de
1 parent 930db25
commit 3c090de
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/nodes.py b/nodes.py
@@ -3,7 +3,6 @@
 import cv2
 import numpy as np
 import torch
-import torchvision.transforms as transforms
 
 from .PixelOE.pixeloe import pixelize
 
@@ -84,15 +83,9 @@ def INPUT_TYPES(cls):
 
     def process(self, img, mode, target_size, patch_size, thickness, color_matching, contrast, saturation, colors, no_upscale, no_downscale):
 
-        # Convert image from PyTorch tensor to NumPy array
         img = img.squeeze().numpy()
-
-        # Convert image from float32 to uint8
         img = (img * 255).astype(np.uint8)
-
-        # Convert image from RGB to BGR
         img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-        print(img.shape)
 
         img_pix = pixelize(img, mode, target_size, patch_size, thickness, color_matching, contrast, saturation, colors, no_upscale, no_downscale)
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,3 @@
 torch
-torchvision
 opencv-python
 numpy