ManglerFTW / ComfyI2I

ComfyI2I is a set of custom nodes for ComfyUI that help with image 2 image functions.
Other
156 stars 15 forks source link

[Feature Request] Color Transfer for Video #14

Open zae-park opened 11 months ago

zae-park commented 11 months ago

I'm pleased to have discovered this project.

Currently, I am working on utilizing AnimateDiff to generate a video from a given image, such as the stainless cup in the GIF provided below. I intend to modify the background color (fog in the GIF) using the 'Color Transfer' feature within this project.

imagegen

The expected behavior of 'Color Transfer' is to take a source video and a target video composed of an image sequence and a single mask image (or a mask image sequence) as inputs and produce a color-transferred image sequence as the output.

However, the output of 'Color Transfer' currently results in a single color-transferred image.

My suspicion is that this problem may stem from the ColorXfer2 method within the Color_Correction class, as it seems to eliminate the batch dimension of the mask tensor.

I'm uncertain about the clever solution to this, but I believe it could potentially be resolved by preserving the batch dimension.

zae-park commented 11 months ago

I think this is not a clever solution, but i still created a PR.