├── LICENSE ├── README.md ├── _config.yml ├── args_parser.py ├── convex_loss.py ├── data_utils ├── ShapeNetDataLoader.py ├── __init__.py └── meta │ ├── anno_paths.txt │ └── class_names.txt ├── environment.yml ├── fitting.py ├── models ├── pointnet.py ├── pointnet2_cls_msg.py ├── pointnet2_cls_ssg.py ├── pointnet2_part_seg_msg.py ├── pointnet2_part_seg_ssg.py ├── pointnet2_sem_seg.py ├── pointnet_cls.py ├── pointnet_part_seg.py ├── pointnet_sem_seg.py ├── pointnet_util.py ├── pointnet_utils.py ├── pretrain_pointnet2_part_seg_msg.py └── reconstruction.py ├── pretrain_partseg_shapenet.py ├── provider.py ├── snip.PNG ├── src ├── VisUtils.py ├── __init__.py ├── augment_utils.py ├── color_utils.py ├── dgcnn.py ├── ellipsoid_fitting.py ├── ellipsoid_fitting_numpy.py ├── ellipsoid_utils.py ├── eval_utils.py ├── fitting_utils.py ├── guard.py ├── mean_shift.py ├── sample_ellipsoid.py └── utils.py ├── testing.py └── train_partseg_shapenet.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/README.md -------------------------------------------------------------------------------- /_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/_config.yml -------------------------------------------------------------------------------- /args_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/args_parser.py -------------------------------------------------------------------------------- /convex_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/convex_loss.py -------------------------------------------------------------------------------- /data_utils/ShapeNetDataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/data_utils/ShapeNetDataLoader.py -------------------------------------------------------------------------------- /data_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_utils/meta/anno_paths.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/data_utils/meta/anno_paths.txt -------------------------------------------------------------------------------- /data_utils/meta/class_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/data_utils/meta/class_names.txt -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/environment.yml -------------------------------------------------------------------------------- /fitting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/fitting.py -------------------------------------------------------------------------------- /models/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet.py -------------------------------------------------------------------------------- /models/pointnet2_cls_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet2_cls_msg.py -------------------------------------------------------------------------------- /models/pointnet2_cls_ssg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet2_cls_ssg.py -------------------------------------------------------------------------------- /models/pointnet2_part_seg_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet2_part_seg_msg.py -------------------------------------------------------------------------------- /models/pointnet2_part_seg_ssg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet2_part_seg_ssg.py -------------------------------------------------------------------------------- /models/pointnet2_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet2_sem_seg.py -------------------------------------------------------------------------------- /models/pointnet_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet_cls.py -------------------------------------------------------------------------------- /models/pointnet_part_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet_part_seg.py -------------------------------------------------------------------------------- /models/pointnet_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet_sem_seg.py -------------------------------------------------------------------------------- /models/pointnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet_util.py -------------------------------------------------------------------------------- /models/pointnet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pointnet_utils.py -------------------------------------------------------------------------------- /models/pretrain_pointnet2_part_seg_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/pretrain_pointnet2_part_seg_msg.py -------------------------------------------------------------------------------- /models/reconstruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/models/reconstruction.py -------------------------------------------------------------------------------- /pretrain_partseg_shapenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/pretrain_partseg_shapenet.py -------------------------------------------------------------------------------- /provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/provider.py -------------------------------------------------------------------------------- /snip.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/snip.PNG -------------------------------------------------------------------------------- /src/VisUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/VisUtils.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/augment_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/augment_utils.py -------------------------------------------------------------------------------- /src/color_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/color_utils.py -------------------------------------------------------------------------------- /src/dgcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/dgcnn.py -------------------------------------------------------------------------------- /src/ellipsoid_fitting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/ellipsoid_fitting.py -------------------------------------------------------------------------------- /src/ellipsoid_fitting_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/ellipsoid_fitting_numpy.py -------------------------------------------------------------------------------- /src/ellipsoid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/ellipsoid_utils.py -------------------------------------------------------------------------------- /src/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/eval_utils.py -------------------------------------------------------------------------------- /src/fitting_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/fitting_utils.py -------------------------------------------------------------------------------- /src/guard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/guard.py -------------------------------------------------------------------------------- /src/mean_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/mean_shift.py -------------------------------------------------------------------------------- /src/sample_ellipsoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/sample_ellipsoid.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/src/utils.py -------------------------------------------------------------------------------- /testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/testing.py -------------------------------------------------------------------------------- /train_partseg_shapenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hippogriff/prifit/HEAD/train_partseg_shapenet.py --------------------------------------------------------------------------------