/ECE371Q_FinalProject_Colorize

Primary LanguagePython

IMAGE COLORIZATION USING CONVOLUTIONAL AUTOENCODER

Final project for ECE 371Q - Digital Image Processing

peyto lake

Left: original, Middle: grayscale input, Right: colorized output (from training validation)

grad photo

Left: original, Middle: model output after epoch 1, Right: output after epoch 16

training loss

Training loss after 16 epochs

autoencoder

Autoencoder model