diff --git a/README.md b/README.md index 065380a..ae937b5 100644 --- a/README.md +++ b/README.md @@ -2,6 +2,13 @@ Implementation of [_Let there be Color!_](http://iizuka.cs.tsukuba.ac.jp/projects/colorization/en/) by Satoshi Iizuka, Edgar Simo-Serra and Hiroshi Ishikawa. +_Colorized 100 years old photo of Książ Castle:_ + +![Colorized Książ Castle, Poland](colorized/ksiaz-castle.png "Colorized Książ Castle, Poland") + +_Other colorized images:_ + +![Colorized images](colorized/results.png "Sample of colorized images") ### First run [Places365-Standard](http://places2.csail.mit.edu/download.html) @@ -16,7 +23,10 @@ $ make split ``` ### Requirements -Python (3.6.3), pytorch (0.4.1), torchvision (0.2.1), skimage (0.14.1), numpy (1.15.2), Jupyter Notebook(4.4.0) +Code is written in Python 3.6.3. [Here](requirements.txt) are all requirements, hit the following to install: +```bash +pip3 install -r requirements.txt +``` ### Network training diff --git a/colorize.py b/colorize.py index 4590d61..f233365 100644 --- a/colorize.py +++ b/colorize.py @@ -4,7 +4,6 @@ import torch import torchvision.transforms -import matplotlib.pyplot as plt import numpy as np from skimage import io, color diff --git a/colorized/ksiaz-castle.png b/colorized/ksiaz-castle.png new file mode 100644 index 0000000..1aa64a8 Binary files /dev/null and b/colorized/ksiaz-castle.png differ diff --git a/colorized/results.png b/colorized/results.png new file mode 100644 index 0000000..9076fdc Binary files /dev/null and b/colorized/results.png differ diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..cb105c9 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +scikit_image==0.14.1 +torch==0.4.1 +numpy==1.15.2 +torchvision==0.2.1 +skimage==0.0 +PyYAML==4.2b1 +jupyter==1.0.0