Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
WSX c0e159e42c | 3 years ago | |
---|---|---|
yolo4 | 3 years ago | |
LICENSE | 3 years ago | |
README.md | 3 years ago |
this code deploy yolov4 on ROS-Melodic with libtorch1.7.0
环境配置(不唯一):
Ubuntu18.04
CUDA10.2
Libtorch1.7.0-cxx11 ABI
OpenCV3.4.10
torchvision0.8.0
需要修改的地方:
1、在下载好的文件中新建module文件夹,放入你的***.pt模型文件
2、在cmakelists.txt 文件中修改set(CMAKE_PREFIX_PATH "/path/to/libtorch; /path/to/torchvision")
3、在yolo.h文件中修改all_anchors、class_name、smodel_image_size参数; 修改utils.h中的num_classes参数,修改方法代码中有注释
4、修改detect_node.cpp中的modulePath为你的***.pt模型文件路径
测试结果:
RTX2070下,帧率21FPS
No Description
C++ Text
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》