[tutorial] deployment process of yolov5 based on tensorflow Lite

One 、 File conversion ： First the pt File to tflite, I'm based on zldrobit The great god pull Code For file conversion and subsequent deployment . But at first it was about pt I use... To convert files yolov5 The official way , But press zldrobit When the great God's code is deployed to the Android machine, the application will flash back , Exclude phone memory 、 After the application itself , I guess it may be the problem of weight file conversion , Then press again zldrobit The great God changed once , The deployment is complete . Because we often encounter network problems , I sometimes like to manually download and install ：

1、 First download and unzip the project zip Installation package ( I upload the file to the cloud , link ：https://pan.baidu.com/s/1t8jwWtykt11YKpyugGPSjA , Extraction code ：sgt6)

2、 Install environment dependent Libraries ： Enter the project folder , Then execute the command to install the required libraries

pip install -r requirements.txt
pip install tensorflow==2.4.0

3、 File conversion ： Will download yolov5 The weight file of “yolov5s.pt” Put it in weights Folder , And then execute the command ：

python  models/tf.py --weight weights/yolov5s.pt --cfg models/yolov5s.yaml --img 320

In this way, the transformed .tflite file ：

Generated .tflite I also put the file thus , Extraction code xtd1.

4、 Store the converted file in assets Folder , Path as follows ：

 

Two 、 Deploy to the Android end ：

Because I have not installed before java and Android studio, Therefore, it is necessary to install and configure .

（ One ）java Environment configuration  

To download jdk： Download link , And then according to This tutorial You can complete the configuration , But here are two points ：

（1） newly build -> Variable name "JAVA_HOME", A variable's value "D:\Java\jdk-17.0.2"（ namely JDK Installation path for ）

（2） edit —— Variable name "Path"—— Click on " newly build "—— Input "%JAVA_HOME%\bin"—— Click on " determine ", Click Edit again —— Variable name "Path"——" newly build "—— Input "%JAVA_HOME%\jre\bin", Click ok .

（ Two ） install Android studio

Download it on the official website . There are many online installation tutorials , You can refer to it . But personally, I think these tutorials are somewhat complicated , You can download it from the official website and install it all the way .

（ 3、 ... and ） Deploy

1、 open Android studio： I didn't create a shortcut because I like a clean desktop , Can be in Android studio In the installation file of bin Folder open exe Executable files , As shown in the figure below ：

2、 Open the project file ： If you enter... For the first time after installation Android studio Interface , You will see a in the middle of the interface “open” Options , Click and then select the file just downloaded “android” Engineering documents can . But I have already opened this blog several times when I am writing this blog , So let's use my interface to make a simple illustration ：

Open it and wait a little , Because the related dependencies will be loaded automatically , At this time, the next step can be carried out at the same time .

 3、 Open the developer mode of mobile phones ： First use USB Connect your phone to your computer , Then turn on developer mode , Different brands of mobile phones are inspired in different ways , With my experiment oppo For example , Set up —— About mobile phone —— Click the version number continuously 7 Time , You can enter developer mode —— Exit to the setting interface —— Other settings —— Developer options —— Click the open button —— open USB Debug mode , At this time, the mobile phone is in the developer mode .

4、 Get into AS debugging ： Before that , Install your own mobile phone first Android SDK,Tools——SDK Manager——SDK Platforms—— Select the corresponding version according to the version of your mobile phone , My is 8.1, So check 8.1, The details are as follows ：

  Then enter debugging ,Run——Run app—— Choose your own phone , Click on —— Pay attention to the pop-up box —— Operating mobile phone , install , open —— complete , Attached is the final rendering ：

 ps： Some tutorials may include installing drivers , But I didn't install , If the deployment cannot be completed as I have written , Maybe you can try installing .

In the process of learning this deployment, I also found an online model converter , Amway gives you Is my , You can convert multiple files , such as onnx——ncnn, I hope it can help you save compilation time ~

That's all , If you have any questions or mistakes, please send a private letter ~