diff options
| author | Robin Fernandes <robin@soal.org> | 2022-09-23 00:57:42 +0000 |
|---|---|---|
| committer | Robin Fernandes <robin@soal.org> | 2022-09-23 00:57:42 +0000 |
| commit | d26d89377bfb47779daec02e888c3c2c931be1f0 (patch) | |
| tree | e724e90a140ecd2c1481417dd4b66dc6b3daafe7 /modules | |
| parent | 03738668c08852d717b0564fc0c3da89877e52b8 (diff) | |
Remove unnecessary duplication
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/processing.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py index 49fed0d8..d27d86e9 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -20,6 +20,7 @@ import modules.shared as shared import modules.face_restoration
import modules.images as images
import modules.styles
+import logging
# some of those options should not be changed at all because they would break the model, so I removed them from options.
@@ -28,11 +29,13 @@ opt_f = 8 def setup_color_correction(image):
+ logging.info("Calibrating color correction.")
correction_target = cv2.cvtColor(np.asarray(image.copy()), cv2.COLOR_RGB2LAB)
return correction_target
def apply_color_correction(correction, image):
+ logging.info("Applying color correction.")
image = Image.fromarray(cv2.cvtColor(exposure.match_histograms(
cv2.cvtColor(
np.asarray(image),
@@ -367,7 +370,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed: if p.color_corrections is not None and i < len(p.color_corrections):
if opts.save and not p.do_not_save_samples and opts.save_images_before_color_correction:
- images.save_image(Image.fromarray(x_sample), p.outpath_samples, "", seeds[i], prompts[i], opts.samples_format, info=infotext(n, i), p=p, suffix="-before-color-correction")
+ images.save_image(image, p.outpath_samples, "", seeds[i], prompts[i], opts.samples_format, info=infotext(n, i), p=p, suffix="-before-color-correction")
image = apply_color_correction(p.color_corrections[i], image)
if p.overlay_images is not None and i < len(p.overlay_images):
|