diff options
Diffstat (limited to 'modules/img2img.py')
| -rw-r--r-- | modules/img2img.py | 22 |
1 files changed, 0 insertions, 22 deletions
diff --git a/modules/img2img.py b/modules/img2img.py index e3109121..70c99e33 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -1,5 +1,4 @@ import math
-import cv2
import numpy as np
from PIL import Image, ImageOps, ImageChops
@@ -76,18 +75,7 @@ def img2img(prompt: str, negative_prompt: str, prompt_style: str, init_img, init state.job_count = n_iter
- do_color_correction = False
- try:
- from skimage import exposure
- do_color_correction = True
- except:
- print("Install scikit-image to perform color correction on loopback")
-
-
for i in range(n_iter):
- if do_color_correction and i == 0:
- correction_target = cv2.cvtColor(np.asarray(init_img.copy()), cv2.COLOR_RGB2LAB)
-
p.n_iter = 1
p.batch_size = 1
p.do_not_save_grid = True
@@ -101,16 +89,6 @@ def img2img(prompt: str, negative_prompt: str, prompt_style: str, init_img, init init_img = processed.images[0]
- if do_color_correction and correction_target is not None:
- init_img = Image.fromarray(cv2.cvtColor(exposure.match_histograms(
- cv2.cvtColor(
- np.asarray(init_img),
- cv2.COLOR_RGB2LAB
- ),
- correction_target,
- channel_axis=2
- ), cv2.COLOR_LAB2RGB).astype("uint8"))
-
p.init_images = [init_img]
p.seed = processed.seed + 1
p.denoising_strength = min(max(p.denoising_strength * denoising_strength_change_factor, 0.1), 1)
|