HCFN (Pytorch Implementation)

EvalSet-1	EvalSet-2

This is an offical implementation for "Hierarchical Color Fusion Network (HCFN): Enhancing Exemplar-based Video Colorization".

Table of Contents

• Prerequisites
• Getting Started

Prerequisites

• Ubuntu 16.04
• Python 3.6.10
• CPU or NVIDIA GPU + CUDA 10.2 CuDNN
• PyTorch 1.5.1

Getting Started

Installation

• Clone this repository and install virtual environment.

git clone https://github.com/wangyins/HCFN
cd HCFN
conda env create -f environment.yaml

• Download model weights from this link to get "checkpoints.zip"

mkdir -p checkpoints/imagenet/
cd checkpoints/imagenet/
unzip checkpoints.zip

Testing

sh test.sh

Training

sh train.sh