qq_64217766的博客

原创 使用anconda进行YOLOV5-lite的模型训练

因为在树莓派上跑模型，需要加速推理，我们所使用的为onnxruntime的加速，所以需要下载onnxruntime，在树莓派上，输入pip install onnxruntime即可。因为opencv检测的是onnx文件，所以需要转换pt文件，在yolo目录下中的models文件里面，有export.py的代码，将 – weights的这一行改成这样。Cv需要onnx文件，故将pt文件转换成onnx文件，在该目录的export.py是转换的代码，将pt文件搬到该目录下，运行即可。其他版本还没尝试过。