yolov5-6.1 opencv dnn onnx推理python,c++

yolov5 6.1 导出onnx
python export.py --include onnx

python
import cv2
import time
import sys
import numpy as np

def build_model(is_cuda):
net = cv2.dnn.readNet("yolov5s.onnx")
if is_cuda:
print("Attempty to use CUDA")
net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA_FP16)
else:
print("Running on CPU")
net.

yolov5-6.1 opencv dnn onnx推理python,c++最先出现在Python成神之路

版权声明:
作者:Mr李
链接:https://www.techfm.club/p/25305.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>