ChromaGAN
ChromaGAN copied to clipboard
Converted Notebook to Python
Nice colorization model. I was able to replicate your results. I have also converted your notebook to a python version that can be run directly via command line.
Just FYI - I also noticed that there is a issue (probably due to memory leak) in the reconstruction part, it will slow down the inference and eventually crush. I moved that part to a separate function and it is good now.
Hi @Bluesn0w, Thanks for you interest in ChromaGAN.
Can be that you new ChromeGAN.py does the same as the file "ChromaGANPrint.py" that we provided for testing?
Best, Patricia