tensorflow1.1/人脸识别Yale数据集
来源:互联网 发布:帝国cms 火车头 编辑:程序博客网 时间:2024/06/07 03:03
环境tensorflow1.1,python3, matplotlib2.02
Yale人脸数据集共有165张图片,15个人,每个人对应十一张图片
#coding:utf-8"""python 3tensorflow 1.1matplotlib 2.02"""import tensorflow as tfimport scipy.io as sioimport numpy as npimport matplotlib.pyplot as pltlearning_rate = 0.0001batch_size=5#加载数据def read_data(filename): with open(filename,'rb') as f: #记载matlab文件 dict = sio.loadmat(f) return dict['fea'],dict['gnd']train_data,train_labels = read_data('Yale_64x64.mat')#将标签转为0-14train_labels = train_labels-1#shuffle datanp.random.seed(100)train_data = np.random.permutation(train_data)np.random.seed(100)train_labels = np.random.permutation(train_labels)test_data = train_data[0:50,:]test_labels = train_labels[0:50]np.random.seed(200)test_data = np.random.permutation(test_data)np.random.seed(200)test_labels = np.random.permutation(test_labels)#将标签转为one_hot类型def label_to_one_hot(labels_dense, num_classes=15): num_labels = labels_dense.shape[0] index_offset = np.arange(num_labels) * num_classes labels_one_hot = np.zeros((num_labels, num_classes)) labels_one_hot.flat[index_offset + labels_dense.ravel()] = 1 return labels_one_hot#将图片转为灰度图def to4d(img): return img.reshape(img.shape[0],64,64,1).astype(np.float32)/255train_data = to4d(train_data)train_labels = label_to_one_hot(train_labels,15)test_data = to4d(test_data)test_labels = label_to_one_hot(test_labels,15)xs = tf.placeholder(tf.float32,[None,64,64,1])ys = tf.placeholder(tf.float32,[None,15])keep_prob = tf.placeholder(tf.float32)#开始构建卷积神经网络conv1 = tf.layers.conv2d(inputs=xs,filters=32,kernel_size=2,strides=1,padding='same',activation=tf.nn.relu)pool1 = tf.layers.max_pooling2d(conv1,pool_size=2,strides=2)conv2 = tf.layers.conv2d(pool1,filters=72,kernel_size=2,strides=1,padding='same',activation=tf.nn.relu)pool2 = tf.layers.max_pooling2d(conv2,pool_size=2,strides=2)flat = tf.reshape(pool2,[-1,16*16*72])dense = tf.layers.dense(flat,512,tf.nn.relu)dropout = tf.nn.dropout(dense,keep_prob)output = tf.layers.dense(dropout,15)#计算lossloss = tf.losses.softmax_cross_entropy(onehot_labels=ys,logits=output)train = tf.train.AdamOptimizer(learning_rate).minimize(loss)#返回两个参数一个train_opt,一个accaccuracy = tf.metrics.accuracy(labels=tf.argmax(ys,axis=1),predictions=tf.argmax(output,axis=1))[1]with tf.Session() as sess: init = tf.group(tf.global_variables_initializer(),tf.local_variables_initializer()) sess.run(init) for step in range(1000): i = 0 while i < len(train_data): start = i end = i+batch_size batch_x = np.array(train_data[start:end]) batch_y = np.array(train_labels[start:end]) _,c = sess.run([train,loss],feed_dict={xs:batch_x,ys:batch_y,keep_prob:0.75}) i += batch_size if step % 1 ==0: acc = sess.run(accuracy,feed_dict={xs:test_data,ys:test_labels,keep_prob:1}) print('= = = = = = > > > > > > ','step:',step,'loss: %.4f'%c,'accuracy: %.2f' %acc)
数据集:链接:http://pan.baidu.com/s/1jIggvEu 密码:fz15
阅读全文
0 0
- tensorflow1.1/人脸识别Yale数据集
- tensorflow1.1/autoencoder人脸识别
- tensorflow1.1/variational autoencoder人脸识别
- Yale数据集上几个人脸识别算法的结果
- embedding可视化/yale人脸数据集
- tensorflow1.1/构建卷积神经网络人脸识别
- tensorflow1.1/构建双向神经网络识别mnist
- tensorflow1.1/构建卷积神经网络识别文本
- tensorflow1.1/构建深度卷积神经网络识别物体识别
- tensorflow1.1/构建卷积神经网络识别手写数字
- tensorflow1.1/循环神经网络手写数字啊识别
- Yale人脸数据库、PCA算法实现Matlab中特征脸生成和简单的人脸识别
- 人脸识别数据集之MS-Celeb-1M
- 人脸识别数据集 Face Databases
- 人脸识别 数据集 与竞赛
- Yale Open Class 1
- python3/tensorflow1.1
- tensorflow1.1/线性回归
- linux中安装svn
- ACM刷题之hdu————KazaQ's Socks
- 计网复习2
- 正则表达式系列总结
- Python中的编码与格式转换
- tensorflow1.1/人脸识别Yale数据集
- easyui----DataGrid DetailView(数据表格详细展示)
- RecyclerView 获取显示item的位置
- springmvc 最简化配置
- css+js实现进度滚动条
- Win10 .net framework 3.5 安装方法(亲测有效)
- placeholder在IE8中兼容性问题解决
- 腾讯推出新图片格式 TPG,但可能还是没法替代 JPEG
- iOS 应用内付费(IAP)开发步骤