Skip to content

Latest commit

 

History

History
39 lines (33 loc) · 1.21 KB

File metadata and controls

39 lines (33 loc) · 1.21 KB

PFNet-pytorch

PyTorch implementation for reproducing Perspective Network (PFNet) results in the paper Rethinking Planar Homography Estimation Using Perspective Fields by Rui Zeng, Simon Denman, Sridha Sridharan, Clinton Fookes.


COCO Dataset

  • Download the COCO 2014 dataset used in the paper.

Trained Weights.

  • Download our trained weights from Baidu Cloud(Extraction code: enkv) and put it in the root directory

Dependencies

  • Python 3.6
  • Pytorch 1.8.0

Training

  • Train a PFNet model on the COCO dataset from scratch:
    • python train.py --dataset=/path/to/COCO

Evaluation

  • Evaluate the model checkpoint
    • python evaluate.py --dataset=/path/to/COCO --model=./pfnet_0200.pth

Current Performance

MACE: 1.21
results
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

Reference

https://github.com/ruizengalways/PFNet