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
三、檢視影片效果:
- Tensorrt
- 推理幀數25ms
- 影片-安全帽檢測展示
四、程式碼展示:
影片展示公眾號
五、獲得的優勢:
- 支援免費答疑;
- 送搭建好的深度學習環境;
- 整體框架移植性比較好;
- 程式碼結果清晰、詳細註釋;
六、獲取方式:
1.關注下方微信公眾號:
2.點選->商業專案->雜貨鋪 搜尋 322 獲取C++原始碼
轉載請註明出處:
歡迎轉載,註明來源,支援原創,謝謝您。每天認認真真做好每一個專案。歡迎大家關注我的公眾號。