tensorflow 用于实现图片分类

使用tensorflow对自己制作的图片数据集进行分类。 只要将data下使用自己的数据集即可。

import tensorflow as tf
import sys

# change this as you see fit
image_path = sys.argv[1]

# Read in the image_data
image_data = tf.gfile.FastGFile（image_path ‘rb‘）.read（）

# Loads label file strips off carriage return
label_lines = [line.rstrip（） for line 
                   in tf.gfile.GFile（“/Users/dudu/PycharmProjects/tensorflow_image_classifier-master/classifier/retrained_labels.txt“）]

# Unpersists graph from file
with tf.gfile.FastGFile（“/Users/dudu/PycharmProjects/tensorflow_image_classifier-master/classifier/retrained_graph.pb“ ‘rb‘） as f:
    graph_def = tf.GraphDef（）
    graph_def.ParseFromString（f.read（））
    _ = tf.import_graph_def（graph_def name=‘‘）

with tf.Session（） as sess:
    # Feed the image_data as input to the graph and get first prediction
  

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-11-02 14:38  tensorflow_image_classifier-master\
     文件        6148  2017-11-02 14:38  tensorflow_image_classifier-master\.DS_Store
     目录           0  2017-11-02 14:33  tensorflow_image_classifier-master\.idea\
     文件        6148  2017-11-02 14:33  tensorflow_image_classifier-master\.idea\.DS_Store
     目录           0  2017-10-17 15:11  tensorflow_image_classifier-master\.idea\inspectionProfiles\
     文件         211  2017-10-17 15:10  tensorflow_image_classifier-master\.idea\misc.xml
     文件         320  2017-10-17 15:10  tensorflow_image_classifier-master\.idea\modules.xml
     文件         398  2017-10-17 15:11  tensorflow_image_classifier-master\.idea\tensorflow_image_classifier-master.iml
     文件       23587  2017-10-27 12:43  tensorflow_image_classifier-master\.idea\workspace.xml
     文件      181954  2017-10-18 10:39  tensorflow_image_classifier-master\8DE6A0E9-05CB-4599-A3AD-C6A630AC2626.png
     文件      242732  2017-10-18 10:54  tensorflow_image_classifier-master\B8390F27732513598A21DF597A4F74E9.jpg
     文件          63  2017-07-18 07:20  tensorflow_image_classifier-master\bash.sh
     文件          63  2017-07-18 07:20  tensorflow_image_classifier-master\bashDev.sh
     文件          40  2017-07-18 07:20  tensorflow_image_classifier-master\build.sh
     目录           0  2017-11-02 14:38  tensorflow_image_classifier-master\classifier\
     文件        6148  2017-11-02 14:38  tensorflow_image_classifier-master\classifier\.DS_Store
     目录           0  2017-11-02 14:38  tensorflow_image_classifier-master\classifier\Data\
     文件         171  2017-07-18 07:20  tensorflow_image_classifier-master\Dockerfile
     文件         104  2017-07-18 07:20  tensorflow_image_classifier-master\guess.sh
     文件         189  2017-07-18 07:20  tensorflow_image_classifier-master\guessDir.sh
     文件        2281  2017-10-18 10:52  tensorflow_image_classifier-master\README.md
     目录           0  2017-10-18 14:55  tensorflow_image_classifier-master\src\
     文件        6148  2017-11-02 14:34  tensorflow_image_classifier-master\src\.DS_Store
     文件          81  2017-07-18 07:20  tensorflow_image_classifier-master\src\guess.sh
     文件        1293  2017-10-18 14:55  tensorflow_image_classifier-master\src\label_image.py
     目录           0  2017-10-18 14:55  tensorflow_image_classifier-master\src\py\
     文件        2020  2017-10-18 14:55  tensorflow_image_classifier-master\src\py\label_dir.py
     文件         287  2017-07-18 07:20  tensorflow_image_classifier-master\src\train.sh
     目录           0  2017-11-02 14:35  tensorflow_image_classifier-master\testa\
     文件        6148  2017-11-02 14:35  tensorflow_image_classifier-master\testa\.DS_Store
     文件        3445  2017-10-18 14:59  tensorflow_image_classifier-master\testa\elliptical--945.jpg
............此处省略1个文件信息