Osterkarten / modules /cv2win32.py
lllyasviel's picture
i
df0d128
raw
history blame
775 Bytes
import threading
import cv2
flag = 'fooocus_cv2win32'
buffer = []
def worker():
global buffer, flag
while True:
cv2.waitKey(50)
try:
if len(buffer) > 0:
task = buffer.pop(0)
if task is None:
cv2.destroyAllWindows()
else:
img, title = task
cv2.imshow(flag, img)
cv2.setWindowTitle(flag, title)
cv2.setWindowProperty(flag, cv2.WND_PROP_TOPMOST, 1)
except Exception as e:
print(e)
pass
def show_preview(img, title='preview'):
buffer.append((img, title))
def close_all_preview():
buffer.append(None)
threading.Thread(target=worker, daemon=True).start()