Image colorization with ofxTensorFlow2

Just made an example (took a while to understand):

Here is a short description (how to convert the model): example_video_and_image_colorization · Issue #25 · zkmkarlsruhe/ofxTensorFlow2 · GitHub
It uses this pretrained model: Artistia/U-Net-epoch-100-loss-0.006095.hdf5 at main · EnigmAI/Artistia · GitHub
Colorizing a small video (480*360) with 30fps in realtime works quite well (more efficient than the arbitrary style transfer example - but its also a much smaller model)…

2 Likes