宠物识别系统,即给定一张图片,判断图片上的宠物是什么。宠物种类暂定为四类——猫、狗、鼠、兔。之所以想到做这个,是因为在不使用公开数据集的情况下,宠物图片数据集获取的难度相对低一些。
小项目分为如下几个部分:爬虫。从网络上下载宠物图片,构建训练用的数据集。 模型构建、训练和调优。鉴于我们的数据比较少,这部分需要做迁移学习。 模型部署和Web服务。将训练好的模型部署成web接口,并使用Vue.js + Element UI编写测试页面。 好嘞,开搞吧 界面展示:
点击网页中的上传框,我们可以选择图片上传并识别:
当然了,这里不选择我们数据集里的图片更好,哪怕是测试集里的。你可以去网上下载、或者通过其他渠道获取这四种动物的图片来测试,这里我只做演示,就不搞那么麻烦了,直接从数据集里随便选几张照片。我们可以继续上传图片给服务器识别: