首页 飞桨领航团 帖子详情
百度AI达人创造营|第四课、部署与推理方法(一)
收藏
快速回复
飞桨领航团 文章AI达人创造营 1163 0
百度AI达人创造营|第四课、部署与推理方法(一)
收藏
快速回复
飞桨领航团 文章AI达人创造营 1163 0

百度AI达人创造营

  • 课程设计

本节课你可以学习到以下的内容:从0到1全流程部署口罩检测模型:从自定义数据集、数据集划分、模型训练、模型导出、安卓端部署,实现一个自己的安卓demo。

  • 开始上课

一、 项目简介

1、该项目使用Labelimg进行数据标注,自定义数据集;
2、使用Paddlex将数据集划分为训练集、测试集;
3、使用PaddleDetection目标检测套件训练模型;
4、最后导出模型,通过PaddleLite生成.nb文件,部署到手机上;
5、安卓部署详细操作;
(1)配置系统环境
(2)下载PaddleLite-dome
(3)下载Android Studio开发环境并配置软件环境
(4)dome调试
(5)根据dome,加入自己的模型,修改配置,实现自己的dome,并调试
(6)将APP打包成可安装程序.apk文件
二 、数据标注

数据标注在前面的第二节课中已经讲过了。https://aistudio.baidu.com/paddle/forum/topic/show/989679

三 、paddlex划分数据集

利用paddle X划分数据集:

!paddlex --split_dataset --format VOC --dataset_dir masks/VOC_MASK --val_value 0.2 --test_value 0.1

四 PaddleDetection目标检测套件使用,进行模型的创建、训练、导出

1、PaddleDetection下载、配置、数据集文件的处理

2、修改模型文件

这里在第三节课程中讲过了,并且为第三次作业中练习过。

https://aistudio.baidu.com/paddle/forum/topic/show/989786

3、模型训练

4、模型预测效果展示

5、模型导出

!python tools/export_model.py -c configs/ssd/ssd_mobilenet_v1_300_120e_voc.yml -o weights=output/ssd_mobilenet_v1_300_120e_voc/best_model.pdparams --output_dir ./inference

五、PaddleLite生成.nb模型文件

PaddleLite文档

https://paddle-lite.readthedocs.io/zh/latest/demo_guides/android_app_demo.html

六、安卓端部署

1、准备Android Studio开发环境(安装系统为Ubuntu 64-bit,Win用户可以安装虚拟机进行操作)

2、导入Paddle-Lite-Demo

3、配置SDK和NDK

4、模型和标签的拷贝

5、修改项目文件

6、准备手机

7、效果展示

8、apk打包

七、总结

本次部署选择object_detection_demo是考虑到有的小伙伴想做其它方面的检测,给大家介绍一个通用的检测demo,若专门做口罩检测可选择mask_detection_demo。

0
收藏
回复
在@后输入用户全名并按空格结束,可艾特全站任一用户