search

WeijingShi / Point-GNN

Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloud, CVPR 2020.
MIT License
523 stars 114 forks source link

Inference on CPU #76

Open abhigoku10 opened 3 years ago

abhigoku10 commented 3 years ago

@WeijingShi hi thanks for sharing the code base i have few queries

  1. can we run the inference code on cpu ? is there any modifications required
  2. Any similar implementation available using pytorch framework Thanks in advance
WeijingShi commented 3 years ago

Hi @abhigoku10, For CPU inference, you can try to simply hide the GPU by CUDA_VISIBLE_DEVICES= python run.py ... There is no plan to migrate code to PyTorch unfortunately. But there are some nice works in this repo (thanks to @Shudeng).