This Python script can turn any black and white or vintage picture into a colorized format. It uses pre-trained models from the richzhang/colorization repository.
This script requires the following packages and pre-trained models to run:
cv2 numpy urllib.request colorization_deploy_v2.prototxt colorization_release_v2.caffemodel
To use the script, simply run the colorizer.py file and provide the path to your black and white or vintage picture as a command line argument. For example:
python colorizer.py path/to/your/picture.jpg
The script will save the colorized version of your picture in the same directory as the original picture, with the suffix _colorized.
Example Here's an example of a black and white picture that has been colorized using this script:
*Coming soon
Credits This script uses pre-trained models from the richzhang/colorization repository. Thank you to the authors for providing these models.