From 689920adcf46b1a490641bcb067c095f46327086 Mon Sep 17 00:00:00 2001 From: aurel-g Date: Sun, 22 Dec 2024 23:53:27 +0100 Subject: [PATCH 1/2] Add requirements.txt --- requirements.txt | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..d928f80 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +opencv-python +kmedoids +deltae +torch_utils +scikit-learn +scikit-image +colour From 5904ebbf1254d1deb0deb2f626e55ea6f116c935 Mon Sep 17 00:00:00 2001 From: aurel-g Date: Sun, 22 Dec 2024 23:53:56 +0100 Subject: [PATCH 2/2] Fix imports problems --- nodes.py | 2 +- utilities/clustering_tests.py | 2 +- utilities/clustering_tests_lab.py | 2 +- utilities/palette_swapping_utils.py | 5 ++--- utilities/scale_utils.py | 2 +- 5 files changed, 6 insertions(+), 7 deletions(-) diff --git a/nodes.py b/nodes.py index 073b5d5..7bf22c9 100644 --- a/nodes.py +++ b/nodes.py @@ -1,6 +1,6 @@ from .comfy_annotations import ComfyFunc, ImageTensor, NumberInput, Choice from .utilities.scale_utils import oe_downscale, downscale -from .utilities.quantization_utils import palette_quantization, palette_swap +from .utilities.quantization_utils import palette_quantization from .utilities.torch_utils import tensor2pil, pil2tensor from .utilities.palette_swapping_utils import palette_swap diff --git a/utilities/clustering_tests.py b/utilities/clustering_tests.py index 472edef..eac2d7c 100644 --- a/utilities/clustering_tests.py +++ b/utilities/clustering_tests.py @@ -23,7 +23,7 @@ OPTICS, SpectralClustering, ) -from correction_utils import ( +from .correction_utils import ( brightness_correction_michelson, brightness_correction_std, contrast_correction_michelson, diff --git a/utilities/clustering_tests_lab.py b/utilities/clustering_tests_lab.py index ad7f253..7d6fe00 100644 --- a/utilities/clustering_tests_lab.py +++ b/utilities/clustering_tests_lab.py @@ -24,7 +24,7 @@ OPTICS, SpectralClustering, ) -from correction_utils import ( +from .correction_utils import ( brightness_correction_michelson, brightness_correction_std, contrast_correction_michelson, diff --git a/utilities/palette_swapping_utils.py b/utilities/palette_swapping_utils.py index cb336ec..b673135 100644 --- a/utilities/palette_swapping_utils.py +++ b/utilities/palette_swapping_utils.py @@ -8,7 +8,7 @@ import torch import kornia from tqdm import tqdm -from correction_utils import ( +from .correction_utils import ( brightness_correction_michelson, contrast_correction_michelson, contrast_correction_std, @@ -19,8 +19,7 @@ ) import torch_utils import skimage -import palette_swapping_utils -import scale_utils +from .scale_utils import * from PIL import Image from itertools import product from PIL import ImageDraw diff --git a/utilities/scale_utils.py b/utilities/scale_utils.py index 28493ef..243af81 100644 --- a/utilities/scale_utils.py +++ b/utilities/scale_utils.py @@ -4,7 +4,7 @@ from itertools import product import cv2 -from oe_utils import outline_expansion, match_color +from .oe_utils import outline_expansion, match_color def k_centroid(image: Image, downscale_factor: int, centroids=2) -> Image: