From c08c8583742222bb3a1f94c88a11e93075502a71 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 4 Aug 2024 16:35:30 +0300 Subject: [PATCH] don't require cv2 for everything --- nodes/image_nodes.py | 6 +++++- nodes/mask_nodes.py | 5 ++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/nodes/image_nodes.py b/nodes/image_nodes.py index 165831a..994059b 100644 --- a/nodes/image_nodes.py +++ b/nodes/image_nodes.py @@ -8,7 +8,11 @@ import os import re import json import hashlib -import cv2 +try: + import cv2 +except: + print("OpenCV not installed") + pass from PIL import ImageGrab, ImageDraw, ImageFont, Image, ImageSequence, ImageOps from nodes import MAX_RESOLUTION, SaveImage diff --git a/nodes/mask_nodes.py b/nodes/mask_nodes.py index bb2065e..b88d1a5 100644 --- a/nodes/mask_nodes.py +++ b/nodes/mask_nodes.py @@ -5,7 +5,10 @@ from PIL import Image, ImageDraw, ImageFilter, ImageFont import scipy.ndimage import numpy as np -import matplotlib.pyplot as plt +try: + import matplotlib.pyplot as plt +except: + print("Matplotlib not installed") from contextlib import nullcontext import os