chrischoy/pytorch_knn_cuda

Stars
67
Rank 462,689 (Top 10 %)
Language Cuda
Created about 7 years ago
Updated about 5 years ago

chrischoy/pytorch_knn_cuda

chrischoy

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

K-Nearest Neighbor in Pytorch

3D-R2N2

Single/multi view image(s) to voxel reconstruction using a recurrent neural network

FCGF

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

pytorch-custom-cuda-tutorial

Tutorial for building a custom CUDA function for Pytorch

DeepGlobalRegistration

[CVPR 2020 Oral] A differentiable framework for 3D registration

SpatioTemporalSegmentation

4D Spatio-Temporal Semantic Segmentation on a 3D video (a sequence of 3D scans)

fully-differentiable-deep-ndf-tf

Fully differentiable deep-neural decision forest in tensorflow

MakePytorchPlusPlus

How and why you want to make your pytorch CUDA/CPP extension with a Makefile

knn_cuda

Fast K-Nearest Neighbor search with GPU

open-ucn

The first fully convolutional metric learning for geometric/semantic image correspondences.

HighDimConvNets

[CVPR 2020 Oral] High-dimensional Convolutional Networks for Geometric Pattern Recognition

gesvd

Pytorch extension for Singular Value Decompostion (SVD) with LAPACK gesvd backend

SUN_RGBD

Reorganized SUN RGBD dataset

SpatioTemporalSegmentation-ScanNet

enriching_object_detection

CUDA-FFT-Convolution

CUDA FFT convolution

segmentation_lecture

python-venv-setup

Make python virtual environment setup on old servers less painful

MinkowskiEngineBenchmark

mini_lseg

PybindNumpyExample

A simple reference template for pybind11 + numpy

env-setup

Setup my dev environment

dotfiles

torch_spmm