upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplified the way to maintain aspect ratio of the preview, and maintaining aspect ratio of the miniatures

Browse Source
This commit is contained in:
Antoine Buchser 2023-06-05 19:34:38 +01:00 committed by GitHub
parent 4903a7b6fb
commit 75b87f3019
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
roop/ui.py
View File

@ -1,6 +1,6 @@
import tkinter as tk import tkinter as tk
from typing import Any, Callable, Tuple from typing import Any, Callable, Tuple
from PIL import Image, ImageTk from PIL import Image, ImageTk, ImageOps
import webbrowser import webbrowser
from tkinter import filedialog from tkinter import filedialog
from tkinter.filedialog import asksaveasfilename from tkinter.filedialog import asksaveasfilename
@ -69,19 +69,7 @@ def init_slider(frames_count, change_handler):
def update_preview(frame): def update_preview(frame):
img = Image.fromarray(frame) img = Image.fromarray(frame)
width, height = img.size img = ImageOps.contain(img, (max_preview_size, max_preview_size), Image.LANCZOS)
aspect_ratio = 1
if width > height:
aspect_ratio = max_preview_size / width
else:
aspect_ratio = max_preview_size / height
img = img.resize(
(
int(width * aspect_ratio),
int(height * aspect_ratio)
),
Image.ANTIALIAS
)
photo_img = ImageTk.PhotoImage(img) photo_img = ImageTk.PhotoImage(img)
preview_image_frame.configure(image=photo_img) preview_image_frame.configure(image=photo_img)
preview_image_frame.image = photo_img preview_image_frame.image = photo_img
@ -211,7 +199,7 @@ def open_preview_window(get_video_frame, target_path):
def preview_face(path): def preview_face(path):
img = Image.open(path) img = Image.open(path)
img = img.resize((180, 180), Image.ANTIALIAS) img = ImageOps.contain(img, (180, 180), Image.LANCZOS)
photo_img = ImageTk.PhotoImage(img) photo_img = ImageTk.PhotoImage(img)
face_label.configure(image=photo_img) face_label.configure(image=photo_img)
face_label.image = photo_img face_label.image = photo_img
@ -219,7 +207,7 @@ def preview_face(path):
def preview_target(frame): def preview_target(frame):
img = Image.fromarray(frame) img = Image.fromarray(frame)
img = img.resize((180, 180), Image.ANTIALIAS) img = ImageOps.contain(img, (180, 180), Image.LANCZOS)
photo_img = ImageTk.PhotoImage(img) photo_img = ImageTk.PhotoImage(img)
target_label.configure(image=photo_img) target_label.configure(image=photo_img)
target_label.image = photo_img target_label.image = photo_img
@ -312,4 +300,4 @@ def init(
status_label = tk.Label(window, width=580, justify="center", text="Status: waiting for input...", fg="#2ecc71", bg="#2d3436") status_label = tk.Label(window, width=580, justify="center", text="Status: waiting for input...", fg="#2ecc71", bg="#2d3436")
status_label.place(x=10,y=640,width=580,height=30) status_label.place(x=10,y=640,width=580,height=30)
return window return window