fix: add handling for filepaths to image grid (#2414)
previously skipped due to not being in np.ndarray format but string
This commit is contained in:
parent
90839430da
commit
4ea3baff50
@ -22,6 +22,7 @@ def worker():
|
||||
import traceback
|
||||
import math
|
||||
import numpy as np
|
||||
import cv2
|
||||
import torch
|
||||
import time
|
||||
import shared
|
||||
@ -79,16 +80,20 @@ def worker():
|
||||
return
|
||||
|
||||
def build_image_wall(async_task):
|
||||
results = async_task.results
|
||||
results = []
|
||||
|
||||
if len(results) < 2:
|
||||
if len(async_task.results) < 2:
|
||||
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):
|
||||
return
|
||||
if img.ndim != 3:
|
||||
return
|
||||
results.append(img)
|
||||
|
||||
H, W, C = results[0].shape
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user