OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2026-01-15 12:15:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bugfix: Display preview image in gui

Browse Source
This commit is contained in:
torzdf
2022-09-02 12:10:41 +01:00
parent 477e3e2013
commit 8fdb856d05
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/gui/utils.py
View File

@@ -902,8 +902,6 @@ class Images():
try:
logger.debug("Displaying preview: '%s'", img)
size = self._get_current_size(name)
if not size:
return
self._previewtrain[name] = [Image.open(img), None, modified]
self.resize_image(name, size)
self._errcount = 0
@@ -945,7 +943,7 @@ class Images():
name, img.width(), img.height())
return img.width(), img.height()
def resize_image(self, name: str, frame_dims: Tuple[int, int]):
def resize_image(self, name: str, frame_dims: Optional[Tuple[int, int]]) -> None:
""" Resize the training preview image based on the passed in frame size.
If the canvas that holds the preview image changes, update the image size
@@ -955,8 +953,9 @@ class Images():
----------
name: str
The name of the training image to be resized
frame_dims: tuple
The (width (`int`), height (`int`)) of the display panel that will display the preview
frame_dims: tuple, optional
The (width (`int`), height (`int`)) of the display panel that will display the preview.
``None`` if the frame dimensions are not known.
"""
logger.debug("Resizing image: (name: '%s', frame_dims: %s", name, frame_dims)
displayimg = cast(Image.Image, self._previewtrain[name][0])