Merge branch 'main_upstream'

# Conflicts:
#	modules/async_worker.py
This commit is contained in:
Manuel Schmid 2024-03-03 00:46:54 +01:00
commit 0768357136
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 8 additions and 3 deletions

View File

@ -24,6 +24,7 @@ def worker():
import traceback
import math
import numpy as np
import cv2
import torch
import time
import shared
@ -87,16 +88,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