带有TensorRT引擎的YOLOv4
该软件包包含yolov4_trt_node,可使用NVIDIA的TensorRT引擎执行推理
该软件包适用于YOLOv3和YOLOv4。 根据所使用的YOLO模型,相应地更改命令。
搭建环境
安装依赖项
当前环境:
杰特逊Xavier AGX
ROS旋律
Ubuntu 18.04
Jetpack 4.4
TensorRT 7+
依存关系:
OpenCV 3.x版
numpy的1.15.1
Protobuf 3.8.0
皮丘达2019.1.2
onnx 1.4.1(取决于Protobuf)
使用以下命令安装所有依赖项
Install pycuda (takes awhile)
$ cd ${HOME}/catkin_ws/src/yolov4_trt_ros/dependencies
$ ./install_pycuda.sh
In
1