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.
- Host: GitHub
- URL: https://github.com/wazzabeee/image-video-colorization
- Owner: Wazzabeee
- Created: 2023-06-06T17:12:41.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2024-08-31T08:42:24.000Z (about 1 year ago)
- Last Synced: 2025-02-01T12:11:14.632Z (8 months ago)
- Topics: automatic-colorization, cnn, colorization, colorize-image, computer-vision, deep-learning, huggingface-spaces, image-colorization, papers, streamlit, video-colorization, youtube-colorization
- Language: Python
- Homepage: https://medium.com/geekculture/creating-a-web-app-to-colorize-images-and-youtube-videos-80f5be2d0f68
- Size: 5.67 MB
- Stars: 15
- Watchers: 2
- Forks: 2
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
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).