python cnn图像分类_tensorflow 1.0用CNN进行图像分类

tensorflow升级到1.0之后，增加了一些高级模块： 如tf.layers, tf.metrics, 和tf.losses，使得代码稍微有些简化。

任务：花卉分类

版本：tensorflow 1.0

花总共有五类，分别放在5个文件夹下。

闲话不多说，直接上代码，希望大家能看懂：)

复制代码

# -*- coding: utf-8 -*-

from skimage import io,transform

import glob

import os

import tensorflow as tf

import numpy as np

import time

path='e:/flower/'

#将所有的图片resize成100*100

w=100

h=100

c=3

#读取图片

def read_img(path):

cate=[path+x for x in os.listdir(path) if os.path.isdir(path+x)]

imgs=[]

labels=[]

for idx,folder in enumerate(cate):

for im in glob.glob(folder+'/*.jpg'):

print('reading the images:%s'%(im))

img=io.imread(im)

img=transform.resize(img,(w,h))

imgs.append(img)

labels.append(idx)

return np.asarray(imgs,np.float32),np.asarray(labels,np.int32)

data,label=read_img(path)

#打乱顺序

num_example=data.shape[0]

arr=np.arange(num_example)

np.random.shuffle(arr)

data=da