xmm-prio / BEVTrack

This repo is the official implementation of "BEVTrack: A Simple and Strong Baseline for 3D Single Object Tracking in Bird's-Eye View".
MIT License
72 stars 6 forks source link

About the inference speed. #1

Open sanfutian opened 3 months ago

sanfutian commented 3 months ago

Thank you for your great work. I encountered an issue while trying to reproduce this project. In the paper, the inference speed is stated as 200 fps, but during my testing, it's only around 10+ fps, with VoxelNet consuming a significant amount of time. I have installed all the required dependencies as instructed. Below is my system environment:


System environment: sys.platform: linux Python: 3.9.19 (main, May 6 2024, 19:43:03) [GCC 11.2.0] CUDA available: True numpy_random_seed: 482379611 GPU 0,1,2,3,4,5,6,7: NVIDIA GeForce RTX 3090 CUDA_HOME: /usr/local/cuda-11.8 NVCC: Cuda compilation tools, release 11.8, V11.8.89 GCC: gcc (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 PyTorch: 2.0.1 PyTorch compiling details: PyTorch built with:

Runtime environment: cudnn_benchmark: False mp_cfg: {'mp_start_method': 'fork', 'opencv_num_threads': 0} dist_cfg: {'backend': 'nccl'} seed: 482379611 Distributed launcher: none Distributed training: False GPU number: 1

xmm-prio commented 3 months ago

Do you install the dependency of spconv?

sanfutian commented 3 months ago

yes, I install "spconv-cu118 2.3.6"

xmm-prio commented 3 months ago

What's your speed test setup like? The network and data on the gpu?

sanfutian commented 3 months ago

Thank you for responding so quickly! It seems like the previous inference speed was indeed affected by other programs running on the server. In my recent test on an NVIDIA GTX 3090 GPU, the latest inference speeds are as follows: bevtrack(VoxelNet+BMM+head): 14.3ms/frame; pre/postprocess: 7.8ms/frame. It's great to hear that the performance of VoxelNet has significantly improved. Please note that the server is not idle due to the activities of my colleagues' programs. I'll update the latest inference speed under idle conditions.

xmm-prio commented 3 months ago

Wishing you all the best and hoping that BEVTrack could help your further research.