Contextual-Loss-PyTorch icon indicating copy to clipboard operation
Contextual-Loss-PyTorch copied to clipboard

An implement of contextual loss using PyTorch.

Contextual-Loss-PyTorch

###Update Logs

  • 08/02/2019, it can be run on multiple GPUs now.

Introduction

An implement of contextual loss using PyTorch according to Maintaining Natural Image Statistics with the Contextual Loss and The Contextual Loss for Image Transformation with Non-Aligned Data.

Requirements

  • PyTorch >= 1.0.0
  • torchvision (to download the pretrained model of VGG19)