【华为云-上云之路】【每天进步一点点】基于ModelArts，跑通口罩检测案例（Python版本）——完整程序

第一步，我们完成了数据集的创建：https://bbs.huaweicloud.com/forum/thread-58782-1-1.html

第二步，我们完成了模型的训练与转化：https://bbs.huaweicloud.com/forum/thread-59213-1-1.html

接下来，就是跑通程序了，非常简单，基本上分为两部分：第一部分是主函数，用于设计程序总体流程，大体是先读取输入（这里是视频），逐帧送入模型推理，之后将结果保存。第二部分，就是模型推理，为了程序整体整洁，推理是一个单独的模块，主函数调用它来完成初始化和推理任务。这里基于以前发的行人检测的Demo，进行了简单地修改，很快就好了，可见还是要基于现有工程修改，从头开始，自己做不太明智，比较费时间，建议大家开发自己项目的时候，可以在官方例程上二次开发，省时省力哈。

这里就不贴代码了，因为贴了代码不容易发帖成功，容易被拦截，完整的代码、模型和测试视频、最终测试结果都在这里，可自行下载测试：

链接：https://pan.baidu.com/s/1eKtBd4tbxdKmpH7smn8nVA 

提取码：1nl4

下载后，解压，可以得到如下：

其中：

input_video:  输入视频存放路径

output_image: 输出图片存放路径

result_video: 输出图片合成的视频，将图片合成视频的方法，可参考https://bbs.huaweicloud.com/blogs/168952

main.py：执行的主函数

utils.py：是模型推理用到的一些函数

yolo3_resnet18_inference: 模型推理的文件

程序执行非常简单，将包含上述文件的真个文件夹拷贝到Atlas 200 DK上，拷贝文件到开发板的命令可参考：https://bbs.huaweicloud.com/blogs/168928。 之后进入该文件夹，执行mian.py命令即可，具体命令为：

python3 main.py input_video/face_detection_1280_720_25.mp4

这里使用的是Python3，Python2应该也可以，直接将上述命令中的python3换为python，其余不变。

最终效果如图：（红色表示检测到未戴口罩，绿色表示到检测到戴口罩）