写在前面
准备近期将微软的machinelearning-samples翻译成中文,水平有限,如有错漏,请大家多多指正。
如果有朋友对此感兴趣,可以加入我:https://github.com/feiyun0112/machinelearning-samples.zh-cn
图像分类 - 评分示例
问题
图像分类是许多业务场景中的常见情况。 对于这些情况,您可以使用预先训练的模型或训练自己的模型来对特定于自定义域的图像进行分类。
数据集
有两个数据源:tsv文件和图像文件。tsv 文件 包含2列:第一个定义为ImagePath,第二个定义为对应于图像的Label。正如你所看到的,文件没有标题行,看起来像这样:
broccoli.jpg broccoli broccoli.png broccoli canoe2.jpg canoe canoe3.jpg canoe canoe4.jpg canoe coffeepot.jpg coffeepot coffeepot2.jpg coffeepot coffeepot3.jpg coffeepot coffeepot4.jpg coffeepot pizza.jpg pizza pizza2.jpg pizza pizza3.jpg pizza teddy1.jpg teddy bear teddy2.jpg teddy bear teddy3.jpg teddy bear teddy4.jpg teddy bear teddy6.jpg teddy bear toaster.jpg toaster toaster2.png toaster toaster3.jpg toaster训练和测试图像位于assets文件夹中。这些图像属于维基共享资源。
代码演练
解决方案中有一个名为
ImageClassification.Score的项目,它负责以TensorFlow格式加载模型,然后对图像进行分类。ML.NET:模型评分
TextLoader.CreateReader()用于定义将用于在ML.NET模型中加载图像的文本文件的模式。
