Colorization and Style Transfer

Project Machine learning in AI training course Vinbigdata. Working in group to make an end-to-end system to colorize gray image, then apply style transfer.

link model: models

install requirements

pip install -r requirements.txt

run server interminal 1

python main_fastapi.py

run font end in terminal 2

python font_end.py

streamlit run font_end.py

Demo colorization

colorization demo (from top to bottom: gray image, colorized image 1, colorized image 2)

Demo style transfer

style transfer demo (from left to right: style, original image, style transfered image)

Demo video colorization then style transfer

Result

Acknowledge

This project is done by 5-mem group.