C++ yolov5 jetson NX盒子部署安全帽檢測 TensorRT, 支援INT8/FP16

帥小帥發表於2023-03-06

C++ yolov5   jetson  NX盒子部署安全帽檢測 TensorRT, 支援INT8/FP16

安全帽檢測,用C++實現,部署到Nvidia上,tensorrt加速,最新的yolov5部署,支援s,m,l模型和int8,FP16等選擇,而且拿來即用,

全網至此一個

------------

一、環境:

  • ubuntu
  • Jetson nano or Jetson Xavier nx
  • Jetpack 4.5.1
  • python3 with default(jetson nano or jetson xavier nx has default python3 with tensorrt 7.1.3.0 )
  • tensorrt 7.1.3.0
  • torch 1.8.0
  • torchvision 0.9.0
  • torch2trt 0.3.0
  • onnx 1.4.1
  • opencv-python 4.5.3.56
  • protobuf 3.17.3
  • scipy 1.5.4
  •  

備註:免費送博主搭建好的環境,獲取原始碼後可以直接跑;

 

二、yolov5訓練參考:

詳細的訓練過程。

備註:轉載

https://zhuanlan.zhihu.com/p/501798155

 

三、檢視影片效果:

  1. Tensorrt
  2. 推理幀數25ms
  3. 影片-安全帽檢測展示

四、程式碼展示:

影片展示公眾號

五、獲得的優勢:

  1. 支援免費答疑;
  2. 送搭建好的深度學習環境;
  3. 整體框架移植性比較好;
  4. 程式碼結果清晰、詳細註釋;

六、獲取方式

1.關注下方微信公眾號:

2.點選->商業專案->雜貨鋪  搜尋 322   獲取C++原始碼

轉載請註明出處:

歡迎轉載,註明來源,支援原創,謝謝您。每天認認真真做好每一個專案。歡迎大家關注我的公眾號。

相關文章