开启图像分类的实践之路

第一步　创建模型

说时迟，那时快，哪吒已经摩拳擦掌地准备大显身手了！首先，第一个任务是确定模型类型，配置模型基本信息，并记录希望模型实现的功能。

1）打开EasyDL平台主页，网址为https://ai.baidu.com/easydl/。

点击图2-7中的【立即使用】按钮，会显示如图2-8所示的【选择模型类型】选择框，选择模型类型为【图像分类】，进入图2-9所示的操作台界面。

图2-7　EasyDL平台主页

图2-8　选择模型类型

图2-9　操作台界面

2）创建模型。点击操作台页面中的【创建模型】按钮，显示如图2-10所示的界面。在该界面中填写模型名称为“虾蟹分类”，模型归属选择“个人”，填写联系方式、功能描述等信息，点击【完成】按钮，完成模型的创建。

图2-10　创建模型

3）模型创建成功后，就可以在【我的模型】中看到刚刚创建的“虾蟹分类”模型，如图2-11所示。

图2-11　模型列表

第二步　准备数据

这个阶段的主要工作是根据具体图像分类的任务准备相应的数据集，并把数据集上传到EasyDL平台，用来训练模型。

（1）准备数据集

首先，准备用于训练模型的图像数据。对于虾蟹分类任务，我们准备了龙虾和螃蟹两种类型的图像。图片格式均为jpg，除此之外还支持png、bmp、jpeg格式的图片。

然后，将准备好的图像数据按照分类存放在不同的文件夹里，文件夹名称即为图像对应的类别标签（longxia、pangxie）。此处要注意，图像类别名即文件夹名称只能包含字母、数字、下划线，不支持中文命名。

最后，将所有文件夹压缩，命名为xiaxie.zip，压缩包的结构示意图如图2-12所示。

图2-12　压缩包的结构示意图

（2）上传数据集

点击图2-13所示的【数据总览】中的【创建数据集】按钮，进行数据集的创建。如图2-14所示，填写数据集名称及标注类型，点击【完成】按钮，然后在“数据总览”页面中点击【导入】，数据标注状态选择【有标注信息】，导入方式选择【本地导入】并上传压缩包xiaxie.zip，上传完成后点击【确认并返回】按钮，如图2-15和图2-16所示。

图2-13　创建数据集

图2-14　填写数据集信息

图2-15　数据集列表

图2-16　上传数据集

（3）查看数据集

上传成功后，可以在【数据总览】中看到数据集正在导入，如图2-17所示。数据集上传后，需要一段处理时间，大约几分钟后就可以看到数据上传的结果了，如图2-18所示。

图2-17　数据集导入中

图2-18　数据集导入成功

点击【查看与标注】，可以看到图像被分为longxia和pangxie两个标签，如图2-19所示。

第三步　训练模型并校验结果

经过前两步，已经创建好了一个图像分类模型，并且创建了数据集，本步骤的主要任务是用上传的数据训练模型，并且在模型训练完成后，在线校验模型的效果。

图2-19　数据标注

（1）训练模型

经过第二步上传数据成功后，在【我的模型】界面中，点击“虾蟹分类”模型项目的【训练】按钮，在【训练模型】界面中，选择之前创建的图像分类模型，添加分类数据集，就可以开始训练模型。训练时间与数据量有关。这个过程如图2-20、图2-21、图2-22和图2-23所示。

图2-20　点击【训练】按钮

图2-21　添加分类数据集

图2-22　训练模型

图2-23　模型训练中

（2）查看模型效果

模型训练完成后，在【我的模型】列表中可以看到模型效果，如图2-24所示。点击图2-24中的【完整评估结果】，可以看到模型训练的整体情况说明，结果显示该模型的训练效果是比较优异的，如图2-25所示。

图2-24　模型训练结果

图2-25　模型整体评估

（3）校验模型

点击图2-26中的【启动模型校验服务】按钮，大约等待5分钟后，进入【校验模型】界面，如图2-27所示。

图2-26　模型校验

图2-27　校验模型界面

然后，准备一条图像数据，在【校验模型】界面中点击“点击添加图片”按钮添加图像。

最后，使用训练好的模型对上传的图像进行预测，如图2-28所示，成功识别出图像中的物体为螃蟹。

图2-28　校验结果

只见不出半小时，哪吒就把敖丙的三千虾兵蟹将全部辨认清楚了，敖丙一行人惊得目瞪口呆。哪吒昂首阔步地走到敖丙面前，义正词严地说道：“小爷我做到了！回去告诉你父王赶紧给陈塘关施雨，以后休得再兴风作浪！”

“啥？你说啥？”敖丙装傻充愣地环顾四周说道：“你们刚才听到我说什么了吗？”“没有，没有！”虾兵蟹将的嘲笑声不绝于耳。

这可把哪吒气得肝胆欲碎、发指眦裂，他咬牙切齿道：“敖丙，你这无耻小人，不守诺言，为非作歹，无可救药！小爷我今日要给你点颜色看看！”

话音未落，只见敖丙手持方天画戟向哪吒刺来，哪吒纵身一闪，让敖丙扑了个空。敖丙恼羞成怒，示意虾兵蟹将冲上去围攻。哪吒忍无可忍，一把甩出混天绫把三太子牢牢捆住，乾坤圈一出，直接击中了敖丙的头部，瞬间将他打回原形。这一战哪吒锋芒毕露，虾兵蟹将见三太子被打死，吓得魂不附体，连滚带爬地钻回了水里。哪吒见这家伙元神竟是一条通体晶莹剔透的小白龙，想起父亲常年征战，腰带都磨损了，龙筋可是做腰带上好的材料，于是“唰”的一声抽出龙筋，高高兴兴地扬长而去（见图2-29）。

图2-29　哪吒怒抽龙筋