Spaces:

huzey
/

ncut-pytorch

Running on Zero

huzey commited on Nov 11, 2024

Commit

4f504b9

1 Parent(s): bd4e31e

update blend

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4509,7 +4509,7 @@ with demo:
                     ncut_image = ncut_image.resize(original_image.shape[:2][::-1], Image.LANCZOS)
                     ncut_image = np.array(ncut_image)
                     from ncut_pytorch import convert_to_lab_color
-                    ncut_image = convert_to_lab_color(ncut_image)
                     blend_image = 0.5 * original_image + 0.5 * ncut_image
                     blend_image = blend_image.astype(np.uint8)
                     ncut_images[i] = Image.fromarray(blend_image)

                     ncut_image = ncut_image.resize(original_image.shape[:2][::-1], Image.LANCZOS)
                     ncut_image = np.array(ncut_image)
                     from ncut_pytorch import convert_to_lab_color
+                    ncut_image = convert_to_lab_color(ncut_image/255) * 255
                     blend_image = 0.5 * original_image + 0.5 * ncut_image
                     blend_image = blend_image.astype(np.uint8)
                     ncut_images[i] = Image.fromarray(blend_image)