Spaces:
Sleeping
Sleeping
File size: 2,566 Bytes
4a1142a | 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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | import streamlit as st
from PIL import Image
import cv2
import numpy as np
import os
from zipfile import ZipFile
# Title and description of the app
st.title("Image Processing App - Image Augmentations")
st.markdown("""
Upload an image to perform augmentations such as translation, scaling, rotation, cropping, and shearing.
Ten augmented images will be generated, and you can download them.
""")
# Upload an image
uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
if uploaded_image is not None:
# Load and display the uploaded image
img = Image.open(uploaded_image)
img_array = np.array(img)
st.image(img, caption="Uploaded Image", use_column_width=True)
# Prepare the output directory for augmented images
output_dir = "augmented_images"
os.makedirs(output_dir, exist_ok=True)
# Define augmentations
st.write("**Performing augmentations...**")
transformations = [
("Translated", cv2.warpAffine(img_array, np.float32([[1, 0, 50], [0, 1, 50]]), (img_array.shape[1], img_array.shape[0]))),
("Scaled", cv2.resize(img_array, None, fx=1.2, fy=1.2)),
("Rotated", cv2.warpAffine(img_array, cv2.getRotationMatrix2D((img_array.shape[1] // 2, img_array.shape[0] // 2), 45, 1), (img_array.shape[1], img_array.shape[0]))),
("Cropped", img_array[50:img_array.shape[0] - 50, 50:img_array.shape[1] - 50]),
("Sheared", cv2.warpAffine(img_array, np.float32([[1, 0.5, 0], [0.5, 1, 0]]), (img_array.shape[1], img_array.shape[0])))
]
# Apply and save augmentations
augmented_images = []
for i, (name, aug_img) in enumerate(transformations * 2): # Duplicate to generate 10 images
output_path = os.path.join(output_dir, f"aug_{i + 1}_{name}.jpg")
cv2.imwrite(output_path, aug_img)
augmented_images.append(output_path)
st.write("**Augmentation completed!**")
# Display augmented images
st.image([Image.open(img_path) for img_path in augmented_images], caption=[f"Image {i + 1}" for i in range(10)], width=150)
# Create a ZIP file for download
zip_filename = "augmented_images.zip"
with ZipFile(zip_filename, 'w') as zipf:
for file_path in augmented_images:
zipf.write(file_path, os.path.basename(file_path))
# Add a download button for the ZIP file
with open(zip_filename, "rb") as fp:
st.download_button(
label="Download Augmented Images",
data=fp,
file_name="augmented_images.zip",
mime="application/zip"
)
|