lxxue / voxelnet-cls

A dumb implementation of voxelnet in pytorch for object classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

voxelnet-cls

A dumb implementation of voxelnet in pytorch for object classification.

Code mostly based on voxnel-pytorch.

Tested on Ubuntu 16.04 with pytorch 0.3.1(0.4.0+ doesn't work).

Data should be saved as .npy file with shape (None, 1024, 3). Labels should be saved as .npy with shape (None,).

Or you can copy the model and script and then modify them based on your needs.

About

A dumb implementation of voxelnet in pytorch for object classification


Languages

Language:Python 92.1%Language:Shell 5.3%Language:C++ 2.5%