nodes/Fictiverse-ComfyUI_Fictiverse
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Fictiverse
2023-11-22 18:29:49 +01:00
committed by GitHub
parent 37feb7c78a
commit 61b87b18ce
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/FV_NodeGroup_1.py
+11 -5
View File
@@ -1,3 +1,4 @@
from re import S
import cv2
import numpy as np
from skimage.exposure import match_histograms
@@ -6,6 +7,7 @@ from enum import Enum
import torch
import torch.nn.functional as F
from torchvision import transforms
from random import randint
# PIL to Tensor
def pil2tensor(image):
@@ -168,6 +170,8 @@ class DisplaceImageWithDepth: #Modified version of WAS node : https://github.com
"X": ("INT", {"default": 0, "min": -4096, "max": 4096, "step": 1}),
"Y": ("INT", {"default": 0, "min": -4096, "max": 4096, "step": 1}),
"Zoom": ("FLOAT", {"default": 0.0, "min": -1, "max": 1, "step": 0.1}),
"Rotation": ("FLOAT", {"default": 0.0, "min": -90, "max": 90, "step": 1}),
"Shake": ("INT", {"default": 0, "min": 0, "max": 100, "step": 1}),
"LayerCount": ("INT", {"default": 8, "min": 2, "max": 255, "step": 1}),
"Frames": ("INT", {"default": 4, "min": 2, "max": 128, "step": 1}),
},
@@ -178,7 +182,7 @@ class DisplaceImageWithDepth: #Modified version of WAS node : https://github.com
FUNCTION = "displaceImageWithDepth"
CATEGORY = "Fictiverse"
def displaceImageWithDepth(self, Image, Depth, X, Y, Zoom, LayerCount, Frames ):
def displaceImageWithDepth(self, Image, Depth, X, Y, Zoom, Rotation, Shake, LayerCount, Frames ):
Tools = Tools_Class()
@@ -187,14 +191,16 @@ class DisplaceImageWithDepth: #Modified version of WAS node : https://github.com
img = tensor2pil(Image[0])
mask = tensor2pil(Depth[0])
mask = Tools.resize_and_crop(mask, img.size)
shakeX = np.random.randint(low=-100, high=100, size=(Frames,))
shakeY = np.random.randint(low=-100, high=100, size=(Frames,))
fX = X/Frames
fY = Y/Frames
fY = Y/Frames
fZ = Zoom/Frames
for f in range(Frames):
tx = fX * f
ty = fY * f
tx = fX * f + shakeX[f]*(Shake/100)
ty = fY * f + shakeY[f]*(Shake/100)
z = fZ * f
layers, combined = Tools.apply_perspective_transformation(img, mask, tx, ty, z, LayerCount)
result_images.append(pil2tensor(combined))