AndreasLH / Image-Colourization

Image colourization for our project in the DTU Deep Learning course (02456)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Image-Colourization

Image colourization 🖍🎨 for our project in the DTU Deep Learning course (02456)

Milestones

  • Data: use the places365 dataset (remove BW images)
  • Make the baseline (GAN and L1-loss without transfer learning)
  • Test difference between L1 and L2 loss on baseline model
  • 2 backbones VGG19, Xception
  • Quantitative evaluation (colourfulness, peak signal-to-noise ratio (PSNR))
  • Qualitative human evaluation (by us) on 5 images each (discussion in report)
  • Use image labels as additional conditional data and assess improvement
  • Evaluate how image label data improved the model

How to activate on HPC

$ are terminal commands

  1. open terminal in same folder as this project and type the following commands (you can paste them into the terminal with middle mouse click)
  2. $ module load python3/3.9.6
  3. $ module load cuda/11.3
  4. $ python3 -m venv DeepLearning
  5. $ source DeepLearning/bin/activate
  6. $ pip3 install -r requirements.txt

Now everything should be setup. Then see the submit.sh shell script for how it is activated. It should be run from the same path as the project.

About

Image colourization for our project in the DTU Deep Learning course (02456)

License:MIT License


Languages

Language:Jupyter Notebook 99.1%Language:Python 0.8%Language:Shell 0.1%