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

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  //所有的程序都在这里了
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

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

结束！