风火少年战AI:人工智能从编程到实践
上QQ阅读APP看书,第一时间看更新

开启图像分类的实践之路

第一步 创建模型

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

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 哪吒怒抽龙筋