ML.NET 示例:深度学习之集成TensorFlow

 

写在前面

准备近期将微软的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模型中加载图像的文本文件的模式。


                        
关键字:
50000+
5万行代码练就真实本领
17年
创办于2008年老牌培训机构
1000+
合作企业
98%
就业率

联系我们

电话咨询

0532-85025005

扫码添加微信