ptrnn's starred repositories
UnsupervisedPointCloudReconstruction
Experiments on unsupervised point cloud reconstruction.
ShapenetRender_more_variation
A new shapenet rendering 2D image dataset that also contains deph map, normal map and albedo map.
galerkin-transformer
[NeurIPS 2021] Galerkin Transformer: a linear attention without softmax
dti-sprites
(ICCV 2021) Code for "Unsupervised Layered Image Decomposition into Object Prototypes" paper
TopographicVAE
Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"
extension-cpp
C++ extensions in PyTorch
2dimageto3dmodel
We evaluate our method on different datasets (including ShapeNet, CUB-200-2011, and Pascal3D+) and achieve state-of-the-art results, outperforming all the other supervised and unsupervised methods and 3D representations, all in terms of performance, accuracy, and training time.
LAP-solvers
Benchmarking Linear Assignment Problem Solvers
differentiable-point-clouds
The reference implementation of "Unsupervised Learning of Shape and Pose with Differentiable Point Clouds"
releasing-research-code
Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)
Guided-VAE
(CVPR 2020) Guided-VAE: Guided Variational Autoencoder for Disentanglement Learning
HungarianGPU
An GPU/CUDA implementation of the Hungarian algorithm