Skip to content

Latest commit

 

History

History
26 lines (14 loc) · 946 Bytes

README.md

File metadata and controls

26 lines (14 loc) · 946 Bytes

Movie Colorizer

This model contains a deep learning model that adds color to black-and-white movies. The predictor builds on the Pix2Pix neural network architecture.

Alt text

(Figure 1. Color predictions for black-and-white movie frames. Predictions were yellowish in general, but some frames were astonishing. The movie in context is Kansas City Confidential, released in 1952.)

Pix2Pix implementation found in model/pix2pix.py

Predict colors to a black-and-white movie frame

NOTE: Weights exceeds 100MB. Therefore, they are not uploaded in this repository. Send a message to majdj@kth.se to obtain them. When you get them, place the .h5 file in data/weights/

  • Navigate to the repository

  • Place your .jpg image in data/test/Y

  • Run,

python3 main.py --predict --test_path 'data/test/Y/[name_of_your_file].jpg'
  • Navigate to data/result/frame_prediction to obtain the result