mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: add handling for filepaths to image grid (#2414)

Browse Source 
previously skipped due to not being in np.ndarray format but string
This commit is contained in:
Manuel Schmid 2024-03-03 00:21:59 +01:00 committed by GitHub
parent 90839430da
commit 4ea3baff50
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/async_worker.py
View File

@ -22,6 +22,7 @@ def worker():
import traceback import traceback
import math import math
import numpy as np import numpy as np
import cv2
import torch import torch
import time import time
import shared import shared
@ -79,16 +80,20 @@ def worker():
return return
def build_image_wall(async_task): def build_image_wall(async_task):
results = async_task.results results = []
if len(results) < 2: if len(async_task.results) < 2:
return return
for img in results: for img in async_task.results:
if isinstance(img, str) and os.path.exists(img):
img = cv2.imread(img)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
if not isinstance(img, np.ndarray): if not isinstance(img, np.ndarray):
return return
if img.ndim != 3: if img.ndim != 3:
return return
results.append(img)
H, W, C = results[0].shape H, W, C = results[0].shape