ColorNet__Image-Colorization-Network

A deeping learning beginner's project.
Using Keras and Pytorch to built image colorization network.
Deep Koalarization: Image Colorization using CNNs and Inception-ResNet-v2 (2017).

Keras Version

(1) cd to your local directory
(2) python colorNetMain.py -h
(3)

Keras ColorNet

positional arguments:
path                   path to the dataset (train or predict) category folders are needed
action                 train or predict

optional arguments:
-h, --help             show this help message and exit
--valpath validate_DIR path to the validation dataset
--epoch N              number of total epochs to run
--patience N           stop the epoch earlier when validatation loss doesn't
drop
--resume PATH          path to latest checkpoint (default: none)

To Do List

(1) Complete the prediction part

Pytorch Version

Not Complete Yet!