| import gradio as gr |
| import numpy as np |
| import cv2 |
| import torch |
|
|
| import random |
| import os |
|
|
| import spaces |
| ''' |
| class ImageMorpher: |
| def __init__(self, num_frames=30, fps=10): |
| self.num_frames = num_frames |
| self.fps = fps |
| |
| def morph_images(self, image_A, image_B): |
| """Morphs two images.""" |
| image_A = np.array(image_A) |
| image_B = np.array(image_B) |
| |
| if image_A.shape != image_B.shape: |
| raise ValueError("Images must have the same dimensions.") |
| if image_A.dtype != image_B.dtype: |
| raise ValueError("Images must have the same data type.") |
| |
| height, width, _ = image_A.shape |
| morphed_images = [] |
| |
| for i in range(self.num_frames + 1): |
| alpha = i / self.num_frames |
| beta = 1 - alpha |
| morphed_image = cv2.addWeighted(image_A, alpha, image_B, beta, 0) |
| morphed_images.append(morphed_image) |
| |
| return morphed_images |
| |
| def create_mp4(self, images, output_path="output.mp4"): |
| """Creates an MP4 video from a list of images.""" |
| height, width, _ = images[0].shape |
| fourcc = cv2.VideoWriter_fourcc(*'mp4v') |
| video_writer = cv2.VideoWriter(output_path, fourcc, self.fps, (width, height)) |
| |
| for image in images: |
| video_writer.write(image) |
| |
| video_writer.release() |
| |
| def morph_image_chain(self, images): |
| """Morphs a chain of images and returns the video path.""" |
| morphed_images = [] |
| for i in range(len(images) - 1): |
| morphed_images.extend(self.morph_images(images[i], images[i + 1])) |
| self.create_mp4(morphed_images) |
| return "output.mp4" |
| |
| def swap_to_gallery(images): |
| return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False) |
| |
| def upload_example_to_gallery(images, prompt, style, negative_prompt): |
| return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False) |
| |
| def remove_back_to_files(): |
| return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) |
| |
| # Create an instance of the ImageMorpher class |
| morpher = ImageMorpher() |
| """ |
| # Define the Gradio interface |
| ### Description and style |
| logo = r""" |
| <center><img src='https://photo-maker.github.io/assets/logo.png' alt='PhotoMaker logo' style="width:80px; margin-bottom:10px"></center> |
| """ |
| title = r""" |
| <h1 align="center">🤗 Morphy the morpher 🤗</h1> |
| """ |
| description = r""" |
| <b></b> |
| <br> |
| Morph image chain into mp4 |
| <br> |
| How to use:<br> |
| -Select two or more images using the "select" tool in the "Images" input field.<br> |
| -Click "Submit".<br> |
| |
| The "Output Video" block will display the morphed video, which you can watch and download.<br> |
| """ |
| css = ''' |
| .gradio-container {width: 85% !important} |
| ''' |
| """ |
| with gr.Blocks(css=css) as demo: |
| gr.Markdown(logo) |
| gr.Markdown(title) |
| gr.Markdown(description) |
| |
| |
| with gr.Row(): |
| with gr.Column(): |
| |
| fn=morpher.morph_image_chain, |
| files = gr.File( |
| label="Drag (Select) 1 or more photos of your face", |
| file_types=["image"], |
| file_count="multiple" |
| ) |
| inputs = gr.Gallery(label="Your images",type="numpy", visible=False, columns=5, rows=1, height=200) |
| with gr.Column(visible=False) as clear_button: |
| remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm") |
| with gr.Column(): |
| outputs=gr.Video(label="Output Video",interactive=False), |
| title="Image Chain Morphing App", |
| with gr.Column(): |
| gallery = gr.Gallery(label="Generated Images") |
| |
| files.upload(fn=swap_to_gallery, inputs=files, outputs=[files, clear_button, files]) |
| remove_and_reupload.click(fn=remove_back_to_files, outputs=[files, clear_button, files]) |
| demo.launch() |
| """ |
| |
| import app1 |
| ''' |
| Explanation: |
|
|
| ImageMorpher class: |
|
|
| Encapsulates the morphing logic and video creation. |
| Has num_frames and fps attributes for customization. |
| morph_images method handles morphing two images. |
| create_mp4 method creates the MP4 video. |
| morph_image_chain method: |
| Iterates through the input image list and morphs each pair. |
| Creates the final MP4 video using create_mp4. |
| Returns the path to the video file. |
|
|
|
|
| ''' |
