mirror of
https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git
synced 2026-01-23 13:24:30 +08:00
minor cleanup
This commit is contained in:
parent
022b499c70
commit
91a141c8b8
@ -1,14 +1,12 @@
|
||||
import torch
|
||||
from torchvision import transforms
|
||||
import json
|
||||
from PIL import Image, ImageDraw, ImageFont, ImageColor, ImageFilter, ImageChops
|
||||
from PIL import Image, ImageDraw, ImageFont, ImageFilter
|
||||
import numpy as np
|
||||
from ..utility.utility import pil2tensor, tensor2pil
|
||||
import folder_paths
|
||||
import io
|
||||
import base64
|
||||
|
||||
from comfy.utils import common_upscale
|
||||
|
||||
def parse_color(color):
|
||||
if isinstance(color, str) and ',' in color:
|
||||
|
||||
@ -881,6 +881,7 @@ class CreateVoronoiMask:
|
||||
|
||||
def createvoronoi(self, frames, num_points, line_width, speed, frame_width, frame_height):
|
||||
from scipy.spatial import Voronoi
|
||||
from matplotlib import pyplot as plt
|
||||
# Define the number of images in the batch
|
||||
batch_size = frames
|
||||
out = []
|
||||
|
||||
@ -3,7 +3,6 @@ import torch.nn as nn
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
import json, re, os, io, time
|
||||
import re
|
||||
import importlib
|
||||
|
||||
from comfy import model_management
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user