w111liang222 / lidar-slam-detection

LSD (LiDAR SLAM & Detection) is an open source perception architecture for autonomous vehicle/robotic
Apache License 2.0
457 stars 105 forks source link

无法调用显卡加速1.3.1 #31

Open zhu2900000 opened 3 months ago

zhu2900000 commented 3 months ago

你好,我用的1.3.1版本,切换到目标检测模式后,可输出结果,但是好像没有用到显卡,检测帧率很低(速度慢),请问应该怎么调试以下,我的docker用的gpu版本

Screenshot from 2024-05-27 16-33-53

un server python script dpkg-query: no packages found matching nvidia-l4t-core Running on board: IPC, machine: x86_64, jetpack: 20.04 2024-05-27 08:39:01,478[26853:26853][ infer.py: 23] image object detection is deactived 2024-05-27 08:39:01,490[26853:26853][module_manager.py: 47] pipeline: Source connect to Detect 2024-05-27 08:39:01,491[26853:26853][module_manager.py: 47] pipeline: Detect connect to Sink 2024-05-27 08:39:01,491[26853:26853][module_manager.py: 71] start to setup module: Detect 2024-05-27 08:39:01,506[26853:26853][object_filter.py: 25] class mapping: {'vehicle': 1, 'pedestrian': 2, 'cyclist': 3, 'traffic_cone': 4} 2024-05-27 08:39:01,506[26853:26853][module_manager.py: 73] setup module: Detect, done 2024-05-27 08:39:01,507[26853:26853][module_manager.py: 71] start to setup module: Sink 2024-05-27 08:39:01,513[26853:26853][module_manager.py: 73] setup module: Sink, done 2024-05-27 08:39:01,514[26853:26853][module_manager.py: 71] start to setup module: Source 2024-05-27 08:39:01,514[26874:26874][sink_template.py: 58] deamon UdpSink starts 2024-05-27 08:39:01,523[26853:26853][player_data_manager.py:135] player have 201 frames, duration 19.971323 s /usr/local/lib/python3.8/dist-packages/onnxruntime/capi/onnxruntime_inference_collection.py:65: UserWarning: Specified provider 'OpenVINOExecutionProvider' is not in available provider names.Available providers: 'CPUExecutionProvider' warnings.warn(

w111liang222 commented 3 months ago

v1.3.1版本目标检测在x86上目前只能运行在CPU上(基于openvino),需要自行将onnx文件转换成tensorrt来调用显卡加速

zhu2900000 commented 3 months ago

v1.3.1版本目标检测在x86上目前只能运行在CPU上(基于openvino),需要自行将onnx文件转换成tensorrt来调用显卡加速 谢谢,master分支的支持吗? @w111liang222

w111liang222 commented 3 months ago

master分支是支持的,但对显卡有要求,需要SM80以上架构的,你的tesla P4好像不满足。。

zhu2900000 commented 3 months ago

master分支是支持的,但对显卡有要求,需要SM80以上架构的,你的tesla P4好像不满足。。

谢谢,如果我想让master分支支持p4显卡,应该如何修改代码,有建议吗?

zhu2900000 commented 3 months ago

master分支是支持的,但对显卡有要求,需要SM80以上架构的,你的tesla P4好像不满足。。

谢谢,如果我想让master分支支持p4显卡,应该如何修改代码,有建议吗?

zhu2900000 commented 3 months ago

master分支是支持的,但对显卡有要求,需要SM80以上架构的,你的tesla P4好像不满足。。

谢谢,如果我想让master分支支持p4显卡,应该如何修改代码,有建议吗?

w111liang222 commented 3 months ago

master分支目标检测算法是centerpoint,包含了稀疏卷积,所以这里采用了libspconv这个库(闭源,并且支持运行在RTX30系列以上显卡),如果要在P4上运行,需要修改成使用v1.3.1版本基于pointpillar的检测模型