fungtion / DRCN

Pytorch implementation of Deep Reconstruction Classification Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This is a pytorch implementation of the model Deep Reconstruction-Classification Network for Unsupervised Domain Adapation (DRCN).

Environment

  • Pytorch 0.4.0
  • Python 2.7

Structure

DRCN

Usage

  • put the mnist and svhn data in the entries in dataset, respectively
  • if there is no Grayscale transform in your torchvision, please replace your functional.py and transforms.py with provided files in extra
  • run python main.py for training
  • the trained model will be saved in model, and recontructed images saved in recovery_image
  • In our implementation, no denoising include

Result

real svhn

Real and Recovered SVHN images

About

Pytorch implementation of Deep Reconstruction Classification Networks


Languages

Language:Python 100.0%