search

zhulf0804 / PointPillars

A Simple PointPillars PyTorch Implementation for 3D LiDAR(KITTI) Detection.
MIT License
454 stars 112 forks source link

Evaluation without CUDA #33

Open Shan-handsome opened 1 year ago

Shan-handsome commented 1 year ago

Thank you for your nice implementation. I tried running evaluation without cuda as the command below: python evaluate.py --ckpt pretrained/epoch_160.pth --data_root kitti/ --no_cuda

This command ends up with an error in the function nms_gpu(): TypeError: nms_gpu(): incompatible function arguments. The following argument types are supported:

  1. (arg0: at::Tensor, arg1: at::Tensor, arg2: float, arg3: int) -> int

Is there a cpu version of nms? Or is there any way can fix this error? Thanks.