File size: 1,945 Bytes
e38bf4d d243d0e d2c6d8a d243d0e f42b290 d2c6d8a 014d1cd d243d0e d2c6d8a d243d0e 9b319b3 d2c6d8a b13bb13 d2c6d8a a655cfe d243d0e aec5ec6 d243d0e a19fb9f d2c6d8a a19fb9f d2c6d8a f42b290 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import numpy as np
import gradio as gr
from skimage import io
from PIL import Image
def svd_compress(image_channel, k):
"""Compress a single channel of the image using SVD by keeping only the top k singular values."""
U, S, Vt = np.linalg.svd(image_channel, full_matrices=False)
compressed_channel = np.dot(U[:, :k], np.dot(np.diag(S[:k]), Vt[:k, :]))
return compressed_channel
def process_image(image, k):
"""Process the uploaded image, compress it using SVD for each color channel, and return the result."""
# Convert PIL Image to NumPy array
image_np = np.array(image)
# Separate the RGB channels
if len(image_np.shape) == 3: # Color image
r_channel, g_channel, b_channel = image_np[:, :, 0], image_np[:, :, 1], image_np[:, :, 2]
# Compress each channel using SVD
r_compressed = svd_compress(r_channel, k)
g_compressed = svd_compress(g_channel, k)
b_compressed = svd_compress(b_channel, k)
# Stack the compressed channels back together
compressed_image = np.stack([r_compressed, g_compressed, b_compressed], axis=2)
else: # Grayscale image
compressed_image = svd_compress(image_np, k)
# Clip the values to ensure valid pixel range and convert to PIL Image for output
compressed_image = np.clip(compressed_image, 0, 255)
compressed_image_pil = Image.fromarray(compressed_image.astype(np.uint8))
return compressed_image_pil
# Gradio interface
gr.Interface(fn=process_image,
inputs=[gr.Image(type="pil", label="Upload Image"),
gr.Slider(1, 100, step=1, value=50, label="Compression Rank")],
outputs=gr.Image(type="pil", label="Compressed Image"),
title="Color Image Compression using SVD",
description="Upload an image (color or grayscale) and adjust the compression rank to see the compressed version."
).launch() |