Skip to content

Add requirements.txt and fix imports problems #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Add requirements.txt and fix imports problems #5

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
689920a
Add requirements.txt
aurel-g Dec 22, 2024
5904ebb
Fix imports problems
aurel-g Dec 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion nodes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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

Expand Down
7 changes: 7 additions & 0 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
opencv-python
kmedoids
deltae
torch_utils
scikit-learn
scikit-image
colour
2 changes: 1 addition & 1 deletion utilities/clustering_tests.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
OPTICS,
SpectralClustering,
)
from correction_utils import (
from .correction_utils import (
brightness_correction_michelson,
brightness_correction_std,
contrast_correction_michelson,
Expand Down
2 changes: 1 addition & 1 deletion utilities/clustering_tests_lab.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
OPTICS,
SpectralClustering,
)
from correction_utils import (
from .correction_utils import (
brightness_correction_michelson,
brightness_correction_std,
contrast_correction_michelson,
Expand Down
5 changes: 2 additions & 3 deletions utilities/palette_swapping_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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
Expand Down
2 changes: 1 addition & 1 deletion utilities/scale_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down