Merge branch 'master' of github.com:kainoj/colnet

kainoj · Jan 29, 2019 · 48341ee · 48341ee
2 parents 3706f9d + 9e9c385
commit 48341ee
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 2 deletions.
diff --git 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
@@ -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
diff --git a/colorized/results.png b/colorized/results.png
diff --git a/requirements.txt 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