An open API service indexing awesome lists of open source software.

https://github.com/wazzabeee/image-video-colorization

Colorize your black and white images and YouTube videos for free. Streamlit application based on CNN deployed on Hugging Face.
https://github.com/wazzabeee/image-video-colorization

automatic-colorization cnn colorization colorize-image computer-vision deep-learning huggingface-spaces image-colorization papers streamlit video-colorization youtube-colorization

Last synced: 8 months ago
JSON representation

Colorize your black and white images and YouTube videos for free. Streamlit application based on CNN deployed on Hugging Face.

Awesome Lists containing this project

README

          

# Image Video Colorization
This project was conceived as part of the "Sujets Spéciaux" course at UQAC under the supervision of [Kevin Bouchard](http://www.kevin-bouchard.ca/)

## Objective
The aim was to deploy a video image colorization application. To this end, I have taken up the GitHub project of [Richard Zhang and his co-authors](https://github.com/richzhang/colorization) with models and papers presented at ECCV16 and SIGGRAPH17. Their solutions are based on convolutional neural networks.

The interface for the models was designed on Streamlit and deployed in a [Space on Hugging Face](https://huggingface.co/spaces/Wazzzabeee/image-video-colorization).

The tutorial with all the design steps is available on my [Medium blog](https://medium.com/geekculture/creating-a-web-app-to-colorize-images-and-youtube-videos-80f5be2d0f68).

The following features are available:
- Colorization of a batch of images.
- Colorization of MP4, MKV and AVI video files.
- Colorization of Youtube videos.

## Interface

## Running Locally
If you want to process longer videos and you're limited by the Hugging Face space memory's limits, you can run this app locally.

`ffmpeg.exe` is needed to run this app, you can install it using `brew install ffmpeg` and update the `IMAGEIO_FFMPEG_EXE` environment variable accordingly.

```bash
git clone https://github.com/Wazzabeee/image-video-colorization
cd image-video-colorization
pip install -r requirements.txt
streamlit run 01_📼_Upload_Video_File.py
```

## Todos
Other models based on GANs will probably be implemented in the future if my application for a community grant to gain access to a GPU on Hugging Face is successful.

## References
1. Richard Zhang, Phillip Isola et Alexei A Efros. « Colorful Image Colorization ». In : ECCV. 2016.
2. Richard Zhang et al. « Real-Time User-Guided Image Colorization with Learned Deep Priors ». In : ACM
Transactions on Graphics (TOG) 9.4 (2017).