Pointwise Rotation-Invariant Network
Switch branches/tags
Nothing to show
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md Update README.md Nov 26, 2018
__init__.py add source code Nov 22, 2018
dataset.py update code Dec 7, 2018
hyper.py Update Nov 29, 2018
model.py fix a bug Nov 22, 2018
test.py fix a bug Nov 22, 2018
train.py add source code Nov 22, 2018

README.md

PRIN

Pointwise Rotation-Invariant Network in PyTorch

Overview

This repository is the Pytorch implementation of PRIN (Pointwise Rotation-Invariant Network).

Dependencies

Dataset and pretrained weights

Usage

  • For training, run "python train.py --log_dir log --model_path ./model.py --num_workers 4"
  • For testing, run "python test.py --weight_path ./state.pkl --model_path ./model.py --num_workers 4"

License

MIT

References

Our paper is available on https://arxiv.org/abs/1811.09361.