官方YOLOv5代码:GitHub - ultralytics/yolov5: YOLOv5 in PyTorch > ONNX > CoreML > TFLite
andriod手机不能直接使用pytorch训练的pt模型文件,因此需要转化为tflite模型。因为google针对移动端手机开发了tensorflow-Lite版。
1. 下载zldrobit提供的转化代码:GitHub - zldrobit/yolov5: YOLOv5 in PyTorch > ONNX > CoreML > iOS
2. 创建虚拟环境 conda create -n yolov5_android python==3.7.10 (已有虚拟环境可跳过)
创建虚拟环境可参考:anaconda 安装卸载虚拟环境_苏打水的杯子的博客-CSDN博客_