yoyo-nb / Thin-Plate-Spline-Motion-Model

[CVPR 2022] Thin-Plate Spline Motion Model for Image Animation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is retaining alpha channel poissible?

Inferencer opened this issue · comments

Looking to retain the alpha transparancy of my source image, if that means the output is in a string of png formats that's ok, is this possible?

I have tried .gif etc and found some good results. (I would like a list of supported outputs)
I have also tried convertin to png using the following in demo.py before the last line

import cv2
from skimage import img_as_ubyte
for i in rangle(len(predictions)):
    cv2.imwrite(f'{i}.png', img_as_ubyte(predictions[i][:,:,::-1]))

However I can an error of:

  File "C:\Users\leolo\Thin-Plate-Spline-Motion-Model\demo.py", line 179, in <module>
    for i in rangle(len(predictions)):
NameError: name 'rangle' is not defined

You should use range, not rangle

That has fixed it but it dosent retain alpha transaparancy on the png which is a shame, I'll stick with ffmpeg for splitting as it's 20x faster than whatever TPSMM just tried to do & use a background remover AI.