Skip to content

Unofficial implementation of CrossNet and CrossNet++.

Notifications You must be signed in to change notification settings

Zeqiang-Lai/Crossnet-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CrossNet PyTorch

This repository reimplements the training pipeline of CrossNet and provides an unoffical implemenation of CrossNet++.

The offical implemenation of CrossNet: ECCV2018_CrossNet_RefSR

Reference Papers:

  1. CrossNet: An End-to-end Reference-based Super Resolution Network using Cross-scale Warping
  2. CrossNet++: Cross-scale Large-parallax Warping for Reference-based Super-resolution

Requirements

  • PyTorch
  • Python3 (tested on Python3.7)
  • torchlight

Getting Started

Datasets

Download the original light field datasets, Flower, LFVideo.

Train

python run.py train -s saved/flower/crossnet/base -c configs/flower/crossnet.yaml
python run.py train -s saved/flower/crossnet++/base -c configs/flower/crossnet++.yaml

Test

python run.py test -s saved/flower/crossnet/base -r best
python run.py test -s saved/flower/crossnet++/base -r best

Results

  • Serpate model for each viewpoint. Reference image are at (0,0).
  • Charbonnier loss only
Model Scale ViewPoint PSNR
CrossNet 4 1,1 42.05
3,3
7,7
CrossNet++ 4 1,1
3,3
7,7

Reference

About

Unofficial implementation of CrossNet and CrossNet++.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published