Skip to content

Latest commit

 

History

History
22 lines (16 loc) · 523 Bytes

README.md

File metadata and controls

22 lines (16 loc) · 523 Bytes

simpleview-pytorch

PyTorch implementation of SimpleView from "Revisiting Point Cloud Classification with a Simple and Effective Baseline"

Usage

import torch
from simpleview_pytorch import SimpleView

model = SimpleView(
    num_views=6,
    num_classes=10
)

# depth 'images' generated from n views of point cloud
# [bs, n, c, h, w]
views = torch.randn(1, 6, 1, 128, 128)

model(views) # (1, 10)