实践案例丨基于ModelArts AI市场算法MobileNet_v2实现花卉分类

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

概述

MobileNetsV2是基于一个流线型的架构，它使用深度可分离的卷积来构建轻量级的深层神经网，此模型基于 MobileNetV2: Inverted Residuals and Linear Bottlenecks 中提出的模型结构实现。可以用图像分类任务，比如猫狗分类、花卉分类等等。用户提供一系列带有标注的数据集，该算法会载入在ImageNet-1000上的预训练模型，在用户数据集上做迁移学习。训练后生成的模型可直接在ModelArts平台部署为在线服务或批量服务，同时支持使用CPU、GPU或Ascend 310进行推理。（以上介绍来自ModelArts AI 市场算法介绍）

注意：需要用到ModelArts和OBS桶，建议提前购买相应资源或代金券，或使用免费规格，但OBS是要花钱的。

准备数据集

这里使用的是花卉数据集，共3669张花卉图片，5个种类，数据集下载地址http://download.tensorflow.org/example_images/flower_photos.tgz （数据集来自自博主张震宇的博客中所提供的数据集，在这里感谢大佬，附上大佬博客链接https://bbs.huaweicloud.com/blogs/158092，里面描述的很详细，建议可以看一下，学习学习，以下的上传数据集方法也来自该博客）

下载解压后，进入flower_photos目录，里面有5个子目录

这里要把flower_photos这个文件夹，也就是上面五个文件的上一级文件夹上传到OBS（华为云对象存储服务），建议使用OBS Browser工具上传。OBS Browser下载：https://support.huaweicloud.com/browsertg-obs/obs_03_1003.html

上传好了之后，回到ModelArts首页，点击左侧“数据管理”->“数据集”，然后点击“创建数据集”

然后先在“数据集输入位置”和“数据集输出位置”选择2个OBS上的文件夹（文件夹要自己创建，必须是空文件夹），剩下的部分默认就行，之后点击右下角创建即可。

回到“数据管理”->“数据集”界面，点击刚才创建的数据集

点击数据集，进入数据集界面，选择右上角的导入

点击导入后，选择最开始上传数据集的路径，就是选择flowers_photos文件夹。

选择之后需要等待一会，可能你会看到导入仍然是0，不要着急，等一下。当看到如下情况，表示导入完成了，接下来就可以发布数据集了。

同样，点击数据集，进入刚才导入的界面，选择右上角的发布

这里选择8:2的比例划分训练集和测试集。点击确定即可。等待数据集创建完成。

下面开始训练，这里首先要到AI市场，订阅算法，放心，是免费的，点击链接进入算法订阅 https://console.huaweicloud.com/modelarts/?region=cn-north-4#/aiMarket/aiMarketModelDetail/overview?modelId=0470e6b1-923b-47a8-ba3e-fe93de5ae2a0&type=algo

点击订阅，按照弹出的界面确认即可。