Tensorflow 实现迁移学习的一个简单例子

xiaoxiao2021-02-28  38

Tensorflow 实现迁移学习

实现效果: 运用Inception-V3的网络框架实现花的种类的图片分类。数据

模型

文件结构

-transfer_learning -flower_data //存放原始图片的文件夹,有5个子文件夹, 每个子文件夹的名称为一种花的名称 -daisy //daisy类花图片的文件夹 -dandelion -roses -sunflowers -tulips -LICENSE.txt -model //存放模型的文件夹 -imagenet_comp_graph_label_strings.txt -LICENSE -tensorflow_inception_graph.pb //模型文件 -tmp -bottleneck //保存模型瓶颈层的特征结果 -daisy //daisy类花特征的文件夹 -dandelion -roses -sunflowers -tulips -transfer_flower.py //所有的程序都在这里了

程序 :程序的功能及配置详情见程序。

结束!

转载请注明原文地址: https://www.6miu.com/read-2624144.html

最新回复(0)