Query realted if the model can give RGBA image as output?

Question

Query realted if the model can give RGBA image as output?

roshan2024nar opened this issue 6 months ago · 2 comments

roshan2024nar commented 6 months ago

Hello,

just wanted to know can i train the model for taking jpg or png(Transperent) as input and then get result in png(Transperent).....
how should i modify the code ?

Thank you for the help :)

Answer 1 · 2024-09-12T06:47:58.000Z

You can train png image if you change --input_nc and --output_nc into 4.
And there might be an converting code in loading the data in file "aligned_dataset.py".
You have to delete the convert code of PIL library.
I hope this help.

Answer 2 · 2024-09-12T06:49:37.000Z

Oh, and you might also have to change the normalizing part in file "base_dataset.py"
I remember that the code is for 3 dimension image so you have to add one more dimension when it nomalize it.