yolov5_tensorrt_int8
yolov5_tensorrt_int8 copied to clipboard
TensorRT int8 量化部署 yolov5s 模型,实测3.3ms一帧!
你好,能给一份yolov5s的激活函数替换为relu重新训练后的模型么?
大佬,如何进行动态shape输入?请问你这个repo支持吗
[ 50%] Building CXX object CMakeFiles/YoloV5sEngine.dir/yolov5s_infer.cc.o In file included from /usr/local/opencv-3.4.2/include/opencv2/calib3d.hpp:48:0, from /usr/local/opencv-3.4.2/include/opencv2/opencv.hpp:56, from /home/wangyuanwen/tensorrt_test/yolov5_tensorrt_int8/yolov5s_infer.cc:6: /usr/local/opencv-3.4.2/include/opencv2/features2d.hpp:50:10: fatal error: opencv2/flann/miniflann.hpp: No such file or directory #include "opencv2/flann/miniflann.hpp" ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~ compilation terminated. CMakeFiles/YoloV5sEngine.dir/build.make:62:...
为什么要设置三层的参数?cur_in_tensor_params[0].nDims ? 多谢
netron.start('此处填充简化后的onnx模型路径', port=3344)
make编译错误
(tensorrt) bowen@bowen:~/yolov5/yolov5_tensorrt_int8/build$ make Scanning dependencies of target YoloV5sEngine [ 50%] Building CXX object CMakeFiles/YoloV5sEngine.dir/yolov5s_infer.cc.o /home/bowen/yolov5/yolov5_tensorrt_int8/yolov5s_infer.cc:18:22: warning: ISO C++ forbids converting a string constant to ‘char*’ [-Wwrite-strings] 18 | char* output_name1...
那别的检测算法 仍需要自己写?
如果想要使用trt7.1.2.6的话,其实这个lib是用不了的
文件太大,我就放到百度盘了(链接: https://pan.baidu.com/s/1sF8vZ1JyBvk5Z_IUBP3CgA 密码: qlgk) 操作步骤如下: cd Generate_LibMyTtrEngine-trt721/src vim CMakeLists.txt 设置cuda目录(例如:/usr/local/cuda-11.0/include) make build cd build cmake .. make 在Generate_LibMyTtrEngine-trt721/bin目录下生成libMyTtrEngine-trt721.so动态库
[100%] Linking CXX executable YoloV5sEngine CMakeFiles/YoloV5sEngine.dir/yolov5s_infer.cc.o: In function `main': yolov5s_infer.cc:(.text+0x1de2): undefined reference to `my_init_tensors' yolov5s_infer.cc:(.text+0x1e7b): undefined reference to `my_load_model' yolov5s_infer.cc:(.text+0x21c5): undefined reference to `my_inference_tensors' yolov5s_infer.cc:(.text+0x2610): undefined reference to `my_deinit_tensors' yolov5s_infer.cc:(.text+0x261f):...