新冠肺炎下,地球是一家,不分国籍,不分种族,或者现在只能呆在家中,但是也是一种对抗疫的支持。停课不停学留在家中,不仅是对学生,对于所有人都是有用的。在现阶段,大家可能最需要的不是金钱,而是一个口罩。一个小小的口罩在这次新冠肺炎起到防护作用。今天就从一个口罩开始,零代码做一个口罩识别模型。
公有云的好处是降低企业在IT投入的巨大成本外,在开发上还提供各式各样的PaaS / SaaS 让你零代码就可以接入一些需要专业人士才能完成的工作。微软Azure Custom Vision 就是在公有云上提供计算机视觉零代码训练的一个项目。(https://www.customvision.ai/)
01
你需要注册一个Azure账号(https://signup.azure.com/)
注册后你就可以进入Custom Vision的门户,直接可以创建你的计算机视觉项目了。
02
点击New Project,创建一个计算机视觉项目,选择好资源位置(在国内建议用东亚)
这里介绍一下,计算机视觉一般有两种,一种是图像分类,一种是物体识别,简单来说我们要划分图片类型时基本上用分类,而要判断图片里面有什么的时候我们就用物体识别,因为我们需要判断图片里面的人是否有戴口罩,这个时候就是物体识别。当我们选择完物体识别后,我们需要选择场景,建议选用General(compact)通过Custom Vision的模型可以适配到移动设备上。
03
创建成功后,门户上就会出现生成的项目,点击进入你就可以对这个项目进行使用,这里我们首先要去Tags创建一个标签,因为这里是辨识口罩,所以我在这里添加一个口罩的标签mask,按保存。
04
接下来选择Add images 添加数据,这里要注意Custom Vsion只是简化了你不需要写算法的操作,但是前期数据采集还是需要人工的,毕竟人工智能= 人工 + 智能 ,而人工>智能 , 原始数据图片可以通过Bing /Baidu 图片查找,我也整理了在我的Azure Notebook上,文章后面我会把我的Notebook地址分享出去。
05
图片导入后,就需要打标记工作,选择Untagged 标记对每张图片,口罩的位置打标签。
当标签打完后选择tag你可以看到完成的状态,以及对应mask标签的汇总
06
标签完成后,就可以选择Train进行训练,你可以选择不同的场景,因为是例子关系,我用了Quick Training。
训练完成分后,会有相关的一些参数,例如Precision, Recall , mAP, 如果你对结果不满意,可以通过增加数据集的方式,或者调整参数的方式进行多次训练。
可以通过Quick Test的方式进行快速测试
07
如果测试结果满意,你可以把模型下载下来,给自己的不同应用场景使用,选择Performance,然后选择Export 即可导出相关模型。
08
这里我导出ONNX 模型,通过Notebook 展示一下相关代码。
通过Azure Custom Vision你可以零代码完成相关的模型训练,而且模型可以在线导出非常方便,你可以快速搭建一个基于计算机视觉的解决方案,而且你不需要投入很多算例去完成这项工作,节约了资金和团队成本。对于一个没有AI基础的人或者团队,是非常好的工具。大家一起动手试试吧。
相关材料
附上我的Azure Notebook : (https://customvisionmaskdemo-kinfey.notebooks.azure.com/j/notebooks/sample01.ipynb)
备注一些线上文档:
(https://docs.microsoft.com/en-us/azure/cognitive-services/custom-vision-service/)