如何用Tensorflow框架构建用于食品分类的机器学习模型？

如何用Tensorflow框架构建用于食品分类的机器学习模型？

摘要

在今天的报告中，我们将分析食品以预测它们是否可以食用。我们应用最先进的迁移学习方法和Tensorflow框架来构建用于食品分类的机器学习模型。

介绍

图像分类是机器预测图片属于哪个类别的工作。在深度学习开始蓬勃发展之前，图像分类等任务无法达到人类水平。这是因为机器学习模型无法确定图像的邻域知识。模型只接收像素级命令。由于深度学习的潜力，图像分类任务可以利用被描述为卷积神经网络（CNN）的模型来传递人类水平的性能。CNN是一种研究图像表征的深度学习模型。该模型无需个人参与即可确定从平面到高级的特征。该模型不仅接收像素级别的数据。该模型还通过称为卷积的机制从图像中获取相邻数据。卷积将通过将范围内的像素的编译相乘并将它们相加为一个值来聚合邻域数据。ML模型将接受这些特征以将图片分类为一组。深度学习虽然可以完成人类级别的生产，但需要大量的数据。

如果我们没有它们怎么办？我们可以应用一种称为迁移学习的理论。

执行

步骤1：导入库

我们要求做的第一步是导入库。我们想要TensorFlow、NumPy、os和pandas。如果你还没有修复包，你可以应用pip命令来安装所需的库。请注意，我命令使用的TensorFlow是2．4．1版本，因此请安装该版本。另外，如果你希望采用GPU来训练深度学习模型，请安装11．0版的CUDA，因为该版本包含2．4．1版的TensorFlow。

下面是安装和加载所需库的代码。＃！pipinstalltensorflow＝＝2．4．1＃！pipinstallpandas＃！pipinstallnumpyimportosimportnumpyasnpimportpandasaspdimporttensorflowastf

步骤2：准备数据

正如你在上面看到的，每个文件夹都由图片组成。每个图片文件名包括类别和由下划线分隔的标识符。我们需要产生带有图像文件名的列和带有该文件夹排列的标签的数据框。修复数据集的代码如下所示，defdframe（dtype）：X＝［］y＝［］path＝＇Food－5K／＇＋dtype＋＇／＇foriinos．listdir（path）：＃ImageX．append（i）＃Labely．append（i．split（＇＿＇）［0］）X＝np．array（X）y＝np．array（y）df＝pd．DataFrame（）df［＇filename＇］＝Xdf［＇label＇］＝yreturndfdf＿train＝dframe（＇training＇）df＿val＝dframe（＇validation＇）df＿test＝dframe（＇evaluation＇）df＿train．head（）