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

Change default video quality, Integrate recent directories for UI

Browse Source
This commit is contained in:
henryruhs 2023-06-08 12:58:29 +02:00
parent 60128d3e96
commit b1c2d414ff
2 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
roop/core.py
View File

@ -43,7 +43,7 @@ def parse_args() -> None:
parser.add_argument('--keep-frames', help='keep frames directory', dest='keep_frames', action='store_true', default=False) parser.add_argument('--keep-frames', help='keep frames directory', dest='keep_frames', action='store_true', default=False)
parser.add_argument('--many-faces', help='swap every face in the frame', dest='many_faces', action='store_true', default=False) parser.add_argument('--many-faces', help='swap every face in the frame', dest='many_faces', action='store_true', default=False)
parser.add_argument('--video-encoder', help='adjust output video encoder', dest='video_encoder', default='libx264') parser.add_argument('--video-encoder', help='adjust output video encoder', dest='video_encoder', default='libx264')
parser.add_argument('--video-quality', help='adjust output video quality', dest='video_quality', type=int, default=10) parser.add_argument('--video-quality', help='adjust output video quality', dest='video_quality', type=int, default=18)
parser.add_argument('--max-memory', help='maximum amount of RAM in GB to be used', dest='max_memory', type=int) parser.add_argument('--max-memory', help='maximum amount of RAM in GB to be used', dest='max_memory', type=int)
parser.add_argument('--cpu-cores', help='number of CPU cores to use', dest='cpu_cores', type=int, default=max(psutil.cpu_count() / 2, 1)) parser.add_argument('--cpu-cores', help='number of CPU cores to use', dest='cpu_cores', type=int, default=max(psutil.cpu_count() / 2, 1))
parser.add_argument('--gpu-threads', help='number of threads to be use for the GPU', dest='gpu_threads', type=int, default=8) parser.add_argument('--gpu-threads', help='number of threads to be use for the GPU', dest='gpu_threads', type=int, default=8)

19
roop/ui.py
View File

@ -1,3 +1,4 @@
import os
import tkinter as tk import tkinter as tk
from tkinter import filedialog from tkinter import filedialog
from typing import Callable, Any, Tuple from typing import Callable, Any, Tuple
@ -19,6 +20,9 @@ WINDOW_HEIGHT = 700
WINDOW_WIDTH = 600 WINDOW_WIDTH = 600
PREVIEW_MAX_HEIGHT = 700 PREVIEW_MAX_HEIGHT = 700
PREVIEW_MAX_WIDTH = 1200 PREVIEW_MAX_WIDTH = 1200
RECENT_DIRECTORY_SOURCE = None
RECENT_DIRECTORY_TARGET = None
RECENT_DIRECTORY_OUTPUT = None
def init(start: Callable, destroy: Callable) -> tk.Tk: def init(start: Callable, destroy: Callable) -> tk.Tk:
@ -153,9 +157,11 @@ def update_status(text: str) -> None:
def select_source_path(): def select_source_path():
source_path = filedialog.askopenfilename(title='Select an face image') global RECENT_DIRECTORY_SOURCE
source_path = filedialog.askopenfilename(title='Select an face image', initialdir=RECENT_DIRECTORY_SOURCE)
if is_image(source_path): if is_image(source_path):
roop.globals.source_path = source_path roop.globals.source_path = source_path
RECENT_DIRECTORY_SOURCE = os.path.dirname(roop.globals.source_path)
image = render_image_preview(roop.globals.source_path, (200, 200)) image = render_image_preview(roop.globals.source_path, (200, 200))
source_label.configure(image=image) source_label.configure(image=image)
source_label.image = image source_label.image = image
@ -166,14 +172,17 @@ def select_source_path():
def select_target_path(): def select_target_path():
target_path = filedialog.askopenfilename(title='Select an image or video target') global RECENT_DIRECTORY_TARGET
target_path = filedialog.askopenfilename(title='Select an image or video target', initialdir=RECENT_DIRECTORY_TARGET)
if is_image(target_path): if is_image(target_path):
roop.globals.target_path = target_path roop.globals.target_path = target_path
RECENT_DIRECTORY_TARGET = os.path.dirname(roop.globals.target_path)
image = render_image_preview(roop.globals.target_path) image = render_image_preview(roop.globals.target_path)
target_label.configure(image=image) target_label.configure(image=image)
target_label.image = image target_label.image = image
elif is_video(target_path): elif is_video(target_path):
roop.globals.target_path = target_path roop.globals.target_path = target_path
RECENT_DIRECTORY_TARGET = os.path.dirname(roop.globals.target_path)
video_frame = render_video_preview(target_path, (200, 200)) video_frame = render_video_preview(target_path, (200, 200))
target_label.configure(image=video_frame) target_label.configure(image=video_frame)
target_label.image = video_frame target_label.image = video_frame
@ -184,12 +193,14 @@ def select_target_path():
def select_output_path(start): def select_output_path(start):
global RECENT_DIRECTORY_OUTPUT
if is_image(roop.globals.target_path): if is_image(roop.globals.target_path):
output_path = filedialog.asksaveasfilename(title='Save image output', initialfile='output.png') output_path = filedialog.asksaveasfilename(title='Save image output', initialfile='output.png', initialdir=RECENT_DIRECTORY_OUTPUT)
elif is_video(roop.globals.target_path): elif is_video(roop.globals.target_path):
output_path = filedialog.asksaveasfilename(title='Save video output', initialfile='output.mp4') output_path = filedialog.asksaveasfilename(title='Save video output', initialfile='output.mp4', initialdir=RECENT_DIRECTORY_OUTPUT)
if output_path: if output_path:
roop.globals.output_path = output_path roop.globals.output_path = output_path
RECENT_DIRECTORY_OUTPUT = os.path.dirname(roop.globals.output_path)
start() start()