Converting the output of net to a colorized image

gustavla / self-supervision

Training and evaluating self-supervised deep neural networks

BSD 3-Clause "New" or "Revised" License

29 stars 6 forks source link

Converting the output of net to a colorized image #5

Open abhaymittal opened 7 years ago

abhaymittal commented 7 years ago

Hi Gustav,

I got the output from the neural net but its a (1,32) vector of h and c values whereas the input image had dimensions (514,514). I compared it with the output of the default neural net in autocolorize and it generates (1,32,512,512) output. Could you please see and provide any changes to the model needed to get a colorized output image?

Regards, Abhay