0
点赞
收藏
分享

微信扫一扫

网络层:地址解析协议ARP

gy2006_sw 2023-06-03 阅读 58

tensorrt源文件路径为/usr/src/tensorrt/bin 下trtexec 可进行onnx转换engine 命令如下:

sudo ./trtexec --onnx=yolov7-tiny.onnx --saveEngine=./yolov7-tiny-fp16.engine --fp16  如果转换fp32 就不加--fp16参数,好像--fp32会报错

include_directories(${PROJECT_SOURCE_DIR}/include)
# include and link dirs of cuda and tensorrt, you need adapt them if yours are different
# cuda
include_directories(/usr/local/cuda-11.4/include)
link_directories(/usr/local/cuda-11.4/lib64)
# tensorrt
include_directories(/usr/include/x86_64-linux-gnu/)
link_directories(/usr/lib/x86_64-linux-gnu/)

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED")

find_package(OpenCV)
include_directories(${OpenCV_INCLUDE_DIRS})

add_executable(yolo ${PROJECT_SOURCE_DIR}/main.cpp yolo.hpp)
target_link_libraries(yolo nvinfer)
target_link_libraries(yolo cudart)
target_link_libraries(yolo ${OpenCV_LIBS})

举报

相关推荐

0 条评论