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