PRS-Net-implementation-PyTorch

This repo implements the PRS-Net proposed by Lin Gao, ICT The access to the paper is: PRS-Net: Planar Reflective Symmetry Detection Net for 3D Models

Proj Structure

• ~root
  • DataPreProc.py (functions to get voxel, point cloud and closest grid for each point)
  • LossFunction.py (two loss functions proposed in the paper)
  • MyDataset.py -Network.py
  • train.py
• utils
  • binvox-rw.py (a file cloned from others' repo, which helps to read .binvox file to numpy array)