Skip to content

Latest commit

 

History

History
 
 

colorization

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

Colorful Image Colorization

Input

Input

(Image above is from https://github.com/richzhang/colorization/tree/master/imgs)

Output

Output

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample image,

$ python3 colorization.py

If you want to specify the input image, put the image path after the --input option.
You can use --savepath option to change the name of the output file to save.

$ python3 colorization.py --input IMAGE_PATH --savepath SAVE_IMAGE_PATH

By adding the --video option, you can input the video.
If you pass 0 as an argument to VIDEO_PATH, you can use the webcam input instead of the video file.

$ python3 colorization.py --video VIDEO_PATH

Reference

Framework

PyTorch

Model Format

ONNX opset=10

Netron

colorizer.onnx.prototxt