import os import numpy as np import matplotlib.pyplot as plt from PIL import Image, ImageChops from skimage import color,data,transform,io #获取所有数据文件夹名称 fileList = os.listdir("F:\\data\\flowers") trainDataList = [] trianLabel = [] testDataList = [] testLabel = [] #读取每一种花的文件 for j in range(len(fileList)): data = os.listdir("F:\\data\\flowers\\"+fileList[j]) #取每一种花四分之一的数据作为测试数据集 testNum = int(len(data)*0.25) #把每种花的图片进行testNum次乱序处理 while(testNum>0): np.random.shuffle(data) testNum -= 1 #把每种花经过乱序后的四分之三当作训练集 trainData = np.array(data[:-(int(len(data)*0.25))]) #把每种花经过乱序后的四分之一当作测试集 testData = np.array(data[-(int(len(data)*0.25)):]) #从上面选出来的训练集中逐张读取出对应的图片 for i in range(len(trainData)): #其中这些图片都要满足jpg格式的 if(trainData[i][-3:]=="jpg"): #读取一张jpg图片 image = io.imread("F:\\data\\flowers\\"+fileList[j]+"\\"+trainData[i]) #把这张图片变成64*64大小的图片 image=transform.resize(image,(64,64)) #保存改变大小的图片到trainDataList列表 trainDataList.append(image) #保存这张图片的标签到trianLabel列表 trianLabel.append(int(j)) #随机生成一个角度，这个角度的范围在顺时针90度和逆时针90度之间 angle = np.random.randint(-90,90) #然后把上面那张64*64大小的图片随机旋转angle个角度 image =transform.rotate(image, angle) #把旋转得到的新图片再变成64*64大小的，因为旋转会改变一张图片的大小 image=transform.resize(image,(64,64)) #把旋转后并且大小是64*64的图片保存到trainDataList列表 trainDataList.append(image) #把旋转后并且大小是64*64的图片对应的标签保存到trianLabel列表 trianLabel.append(int(j)) #逐张读取每种花的测试图片 for i in range(len(testData)): #选取的图片要满足jpg格式的 if(testData[i][-3:]=="jpg"): #读取一张图片 image = io.imread("F:\\data\\flowers\\"+fileList[j]+"\\"+testData[i]) #改变这张图片的大小为64*64 image=transform.resize(image,(64,64)) #把改变后的图片保存到testDataList列表中 testDataList.append(image) #把这张图片对应的标签保存到testLabel列表中 testLabel.append(int(j)) print("图片数据读取完了...")
#打印训练集和测试数据集以及它们对应标签的规模 print(np.shape(trainDataList)) print(np.shape(trianLabel)) print(np.shape(testDataList)) print(np.shape(testLabel))
#保存训练集和测试数据集以及它们对应标签到磁盘 np.save("G:\\trainDataList",trainDataList) np.save("G:\\trianLabel",trianLabel) np.save("G:\\testDataList",testDataList) np.save("G:\\testLabel",testLabel) print("数据处理完了...")
import numpy as np from keras.utils import to_categorical #将训练数据集和测试数据集对应的标签转变为one-hot编码 trainLabel = np.load("G:\\trianLabel.npy") testLabel = np.load("G:\\testLabel.npy") trainLabel_encoded = to_categorical(trainLabel) testLabel_encoded = to_categorical(testLabel) np.save("G:\\trianLabel",trainLabel_encoded) np.save("G:\\testLabel",testLabel_encoded) print("转码类别写盘完了...")
import random import numpy as np trainDataList = np.load("G:\\trainDataList.npy") trianLabel = np.load("G:\\trianLabel.npy") print("数据加载完了...") trainIndex = [i for i in range(len(trianLabel))] random.shuffle(trainIndex) trainData = [] trainClass = [] for i in range(len(trainIndex)): trainData.append(trainDataList[trainIndex[i]]) trainClass.append(trianLabel[trainIndex[i]]) print("训练数据shuffle完了...") np.save("G:\\trainDataList",trainData) np.save("G:\\trianLabel",trainClass) print("训练数据写盘完毕...")
X = np.load("G:\\trainDataList.npy") Y = np.load("G:\\trianLabel.npy") print(np.shape(X)) print(np.shape(Y))
import random import numpy as np testDataList = np.load("G:\\testDataList.npy") testLabel = np.load("G:\\testLabel.npy") testIndex = [i for i in range(len(testLabel))] random.shuffle(testIndex) testData = [] testClass = [] for i in range(len(testIndex)): testData.append(testDataList[testIndex[i]]) testClass.append(testLabel[testIndex[i]]) print("测试数据shuffle完了...") np.save("G:\\testDataList",testData) np.save("G:\\testLabel",testClass) print("测试数据写盘完毕...")
X = np.load("G:\\testDataList.npy") Y = np.load("G:\\testLabel.npy") print(np.shape(X)) print(np.shape(Y)) print(np.shape(testData)) print(np.shape(testLabel))
import tensorflow as tf from random import shuffle INPUT_NODE = 64*64 OUT_NODE = 5 IMAGE_SIZE = 64 NUM_CHANNELS = 3 NUM_LABELS = 5 #第一层卷积层的尺寸和深度 CONV1_DEEP = 16 CONV1_SIZE = 5 #第二层卷积层的尺寸和深度 CONV2_DEEP = 32 CONV2_SIZE = 5 #全连接层的节点数 FC_SIZE = 512 def inference(input_tensor, train, regularizer): #卷积 with tf.variable_scope('layer1-conv1'): conv1_weights = tf.Variable(tf.random_normal([CONV1_SIZE,CONV1_SIZE,NUM_CHANNELS,CONV1_DEEP],stddev=0.1),name='weight') tf.summary.histogram('convLayer1/weights1', conv1_weights) conv1_biases = tf.Variable(tf.Variable(tf.random_normal([CONV1_DEEP])),name="bias") tf.summary.histogram('convLayer1/bias1', conv1_biases) conv1 = tf.nn.conv2d(input_tensor,conv1_weights,strides=[1,1,1,1],padding='SAME') tf.summary.histogram('convLayer1/conv1', conv1) relu1 = tf.nn.relu(tf.nn.bias_add(conv1,conv1_biases)) tf.summary.histogram('ConvLayer1/relu1', relu1) #池化 with tf.variable_scope('layer2-pool1'): pool1 = tf.nn.max_pool(relu1,ksize=[1,2,2,1],strides=[1,2,2,1],padding='SAME') tf.summary.histogram('ConvLayer1/pool1', pool1) #卷积 with tf.variable_scope('layer3-conv2'): conv2_weights = tf.Variable(tf.random_normal([CONV2_SIZE,CONV2_SIZE,CONV1_DEEP,CONV2_DEEP],stddev=0.1),name='weight') tf.summary.histogram('convLayer2/weights2', conv2_weights) conv2_biases = tf.Variable(tf.random_normal([CONV2_DEEP]),name="bias") tf.summary.histogram('convLayer2/bias2', conv2_biases) #卷积向前学习 conv2 = tf.nn.conv2d(pool1,conv2_weights,strides=[1,1,1,1],padding='SAME') tf.summary.histogram('convLayer2/conv2', conv2) relu2 = tf.nn.relu(tf.nn.bias_add(conv2,conv2_biases)) tf.summary.histogram('ConvLayer2/relu2', relu2) #池化 with tf.variable_scope('layer4-pool2'): pool2 = tf.nn.max_pool(relu2,ksize=[1,2,2,1],strides=[1,2,2,1],padding='SAME') tf.summary.histogram('ConvLayer2/pool2', pool2) #变型 pool_shape = pool2.get_shape().as_list() #计算最后一次池化后对象的体积（数据个数\节点数\像素个数） nodes = pool_shape[1]*pool_shape[2]*pool_shape[3] #根据上面的nodes再次把最后池化的结果pool2变为batch行nodes列的数据 reshaped = tf.reshape(pool2,[-1,nodes]) #全连接层 with tf.variable_scope('layer5-fc1'): fc1_weights = tf.Variable(tf.random_normal([nodes,FC_SIZE],stddev=0.1),name='weight') if(regularizer != None): tf.add_to_collection('losses',tf.contrib.layers.l2_regularizer(0.03)(fc1_weights)) fc1_biases = tf.Variable(tf.random_normal([FC_SIZE]),name="bias") #预测 fc1 = tf.nn.relu(tf.matmul(reshaped,fc1_weights)+fc1_biases) if(train): fc1 = tf.nn.dropout(fc1,0.5) #全连接层 with tf.variable_scope('layer6-fc2'): fc2_weights = tf.Variable(tf.random_normal([FC_SIZE,64],stddev=0.1),name="weight") if(regularizer != None): tf.add_to_collection('losses',tf.contrib.layers.l2_regularizer(0.03)(fc2_weights)) fc2_biases = tf.Variable(tf.random_normal([64]),name="bias") #预测 fc2 = tf.nn.relu(tf.matmul(fc1,fc2_weights)+fc2_biases) if(train): fc2 = tf.nn.dropout(fc2,0.5) #全连接层 with tf.variable_scope('layer7-fc3'): fc3_weights = tf.Variable(tf.random_normal([64,NUM_LABELS],stddev=0.1),name="weight") if(regularizer != None): tf.add_to_collection('losses',tf.contrib.layers.l2_regularizer(0.03)(fc3_weights)) fc3_biases = tf.Variable(tf.random_normal([NUM_LABELS]),name="bias") #预测 logit = tf.matmul(fc2,fc3_weights)+fc3_biases return logit
import time import keras import numpy as np from keras.utils import np_utils X = np.load("G:\\trainDataList.npy") Y = np.load("G:\\trianLabel.npy") print(np.shape(X)) print(np.shape(Y)) print(np.shape(testData)) print(np.shape(testLabel)) batch_size = 10 n_classes=5 epochs=16#循环次数 learning_rate=1e-4 batch_num=int(np.shape(X)[0]/batch_size) dropout=0.75 x=tf.placeholder(tf.float32,[None,64,64,3]) y=tf.placeholder(tf.float32,[None,n_classes]) # keep_prob = tf.placeholder(tf.float32) #加载测试数据集 test_X = np.load("G:\\testDataList.npy") test_Y = np.load("G:\\testLabel.npy") back = 64 ro = int(len(test_X)/back) #调用神经网络方法 pred=inference(x,1,"regularizer") cost=tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred,labels=y)) # 三种优化方法选择一个就可以 optimizer=tf.train.AdamOptimizer(1e-4).minimize(cost) # train_step = tf.train.GradientDescentOptimizer(0.001).minimize(cost) # train_step = tf.train.MomentumOptimizer(0.001,0.9).minimize(cost) #将预测label与真实比较 correct_pred=tf.equal(tf.argmax(pred,1),tf.argmax(y,1)) #计算准确率 accuracy=tf.reduce_mean(tf.cast(correct_pred,tf.float32)) merged=tf.summary.merge_all() #将tensorflow变量实例化 init=tf.global_variables_initializer() start_time = time.time() with tf.Session() as sess: sess.run(init) #保存tensorflow参数可视化文件 writer=tf.summary.FileWriter('F:/Flower_graph', sess.graph) for i in range(epochs): for j in range(batch_num): offset = (j * batch_size) % (Y.shape[0] - batch_size) # 准备数据 batch_data = X[offset:(offset + batch_size), :] batch_labels = Y[offset:(offset + batch_size), :] sess.run(optimizer, feed_dict={x:batch_data,y:batch_labels}) result=sess.run(merged, feed_dict={x:batch_data,y:batch_labels}) writer.add_summary(result, i) loss,acc = sess.run([cost,accuracy],feed_dict={x:batch_data,y:batch_labels}) print("Epoch:", '%04d' % (i+1),"cost=", "{:.9f}".format(loss),"Training accuracy","{:.5f}".format(acc*100)) writer.close() print("########################训练结束，下面开始测试###################") for i in range(ro): s = i*back e = s+back test_accuracy = sess.run(accuracy,feed_dict={x:test_X[s:e],y:test_Y[s:e]}) print("step:%d test accuracy = %.4f%%" % (i,test_accuracy*100)) print("Final test accuracy = %.4f%%" % (test_accuracy*100)) end_time = time.time() print('Times：',(end_time-start_time)) print('Optimization Completed')
本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：吴裕雄–天生自然 Tensorflow卷积神经网络：花朵图片识别 - Python技术站
微信扫一扫 
支付宝扫一扫 