big_vision
big_vision copied to clipboard
google-research
Colorization uvim model not working
Hi @andresusanopinto I hope you are well. I tried using the colorization model on my images but out of 4 it colorized only 1 image and the result is also not good at all. Can you please tell me what I am missing ?
Hi @muhammad-ahmed-ghani,
To get colorful samples please set the sampling temperature to 1.0 in the last cell of the colab: temperature = jnp.array(1.0). After fixing the temperature also consider running inference multiple times and then selecting the best colorization attempt. The near zero sampling temperature is not good default for the colorization task, we will likely update the colab soon.
Let us know whether it worked for you.
I changed it but still it adds a slight blueish tint to the images.
That is a bit puzzling, as I am able to get reasonable colorization. If you share a reproduction I may have a look at some point.
That is a bit puzzling, as I am able to get reasonable colorization. If you share a reproduction I may have a look at some point.
You can check the code here. I just changed the images directories.
https://colab.research.google.com/drive/1E5R_pla2cPkwVKxu9cCgvCt9PtAA-PCq?usp=sharing
That is a bit puzzling, as I am able to get reasonable colorization. If you share a reproduction I may have a look at some point.
You can check the code here. I just changed the images directories.
https://colab.research.google.com/drive/1E5R_pla2cPkwVKxu9cCgvCt9PtAA-PCq?usp=sharing
Hi @akolesnikoff, have you find anything on it ?