Skip to content

pytorch implementation of PRNet, with weight transfered

Notifications You must be signed in to change notification settings

liguohao96/pytorch-prnet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-prnet


Introduction

A pytorch implement of PRNet, with weight transferred

Prerequisite

  • Python3
  • tensorflow
  • pytorch
  • skimage(if you want to run with a image)

Usage

> git clone https://github.com/liguohao96/pytorch-prnet.git
> cd pytorch-prnet
> python tf2torch.py --prnet_dir /path/to/prnet/repository

Test

this will train tf_prnet and torch_prnet simultaneously on a random dataset to see if their outputs and loss are the same. (based on my experiement, they are not)

> python test/train.py --prnet_dir /path/to/prnet/repository --step

About

pytorch implementation of PRNet, with weight transfered

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages