#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Wed Jan 18 08:42:55 2017

@author: root
"""

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Jan 16 11:08:21 2017

@author: root
"""
import tensorflow as tf
import frecordfortrain
tf.device(0)
def read_and_decode(filename):
    #根据文件名生成一个队列
    filename_queue = tf.train.string_input_producer([filename])

reader = tf.TFRecordReader()
    _, serialized_example = reader.read(filename_queue)   #返回文件名和文件
    features = tf.parse_single_example(serialized_example,
                                       features={
                                           'label': tf.FixedLenFeature([], tf.int64),
                                           'img_raw' : tf.FixedLenFeature([], tf.string),
                                       })

img = tf.decode_raw(features['img_raw'], tf.uint8)
    img = tf.reshape(img, [227, 227, 3])
 #    img = tf.reshape(img, [39, 39, 3])
    img = tf.cast(img, tf.float32) * (1. / 255) - 0.5
    label = tf.cast(features['label'], tf.int32)
    print img,label
    return img, label
    
def get_batch(image, label, batch_size,crop_size):  
        #数据扩充变换  
    distorted_image = tf.random_crop(image, [crop_size, crop_size, 3])#随机裁剪  
    distorted_image = tf.image.random_flip_up_down(distorted_image)#上下随机翻转  
    #distorted_image = tf.image.random_brightness(distorted_image,max_delta=63)#亮度变化  
    #distorted_image = tf.image.random_contrast(distorted_image,lower=0.2, upper=1.8)#对比度变化  
 
    #生成batch  
    #shuffle_batch的参数:capacity用于定义shuttle的范围,如果是对整个训练数据集,获取batch,那么capacity就应该够大  
    #保证数据打的足够乱  
#    images, label_batch = tf.train.shuffle_batch([distorted_image, label],batch_size=batch_size,  
 #                                                num_threads=16,capacity=50000,min_after_dequeue=10000)  
    images, label_batch = tf.train.shuffle_batch([distorted_image, label],batch_size=batch_size,  
                                                 num_threads=2,capacity=2,min_after_dequeue=10)
    #images, label_batch=tf.train.batch([distorted_image, label],batch_size=batch_size)  
    # 调试显示  
    #tf.image_summary('images', images)  
    print "in get batch"
    print images,label_batch
    return images, tf.reshape(label_batch, [batch_size])    
    
 
#from  data_encoder_decoeder import  encode_to_tfrecords,decode_from_tfrecords,get_batch,get_test_batch  
import  cv2  
import  os  
 
class network(object):  
    def inference(self,images):  
        # 向量转为矩阵  
      #  images = tf.reshape(images, shape=[-1, 39,39, 3])
        images = tf.reshape(images, shape=[-1, 227,227, 3])# [batch, in_height, in_width, in_channels]  
        images=(tf.cast(images,tf.float32)/255.-0.5)*2#归一化处理  
 
 
 
        #第一层  
        conv1=tf.nn.bias_add(tf.nn.conv2d(images, self.weights['conv1'], strides=[1, 4, 4, 1], padding='VALID'),  
                             self.biases['conv1'])  
 
        relu1= tf.nn.relu(conv1)  
        pool1=tf.nn.max_pool(relu1, ksize=[1, 3, 3, 1], strides=[1, 2, 2, 1], padding='VALID')  
 
 
        #第二层  
        conv2=tf.nn.bias_add(tf.nn.conv2d(pool1, self.weights['conv2'], strides=[1, 1, 1, 1], padding='SAME'),  
                             self.biases['conv2'])  
        relu2= tf.nn.relu(conv2)  
        pool2=tf.nn.max_pool(relu2, ksize=[1, 3, 3, 1], strides=[1, 2, 2, 1], padding='VALID')  
 
 
        # 第三层  
        conv3=tf.nn.bias_add(tf.nn.conv2d(pool2, self.weights['conv3'], strides=[1, 1, 1, 1], padding='SAME'),  
                             self.biases['conv3'])  
        relu3= tf.nn.relu(conv3)  
      #  pool3=tf.nn.max_pool(relu3, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='VALID')  
        conv4=tf.nn.bias_add(tf.nn.conv2d(relu3, self.weights['conv4'], strides=[1, 1, 1, 1], padding='SAME'),  
                             self.biases['conv4'])  
        relu4= tf.nn.relu(conv4)
        conv5=tf.nn.bias_add(tf.nn.conv2d(relu4, self.weights['conv5'], strides=[1, 1, 1, 1], padding='SAME'),  
                             self.biases['conv5'])  
        relu5= tf.nn.relu(conv5)
        pool5=tf.nn.max_pool(relu5, ksize=[1, 3, 3, 1], strides=[1, 2, 2, 1], padding='VALID')
        # 全连接层1,先把特征图转为向量  
        flatten = tf.reshape(pool5, [-1, self.weights['fc1'].get_shape().as_list()[0]])  
 
        drop1=tf.nn.dropout(flatten,0.5)  
        fc1=tf.matmul(drop1, self.weights['fc1'])+self.biases['fc1']  
 
        fc_relu1=tf.nn.relu(fc1)  
 
        fc2=tf.matmul(fc_relu1, self.weights['fc2'])+self.biases['fc2']
        fc_relu2=tf.nn.relu(fc2)  
 
        fc3=tf.matmul(fc_relu2, self.weights['fc3'])+self.biases['fc3']
 
        return  fc3

def __init__(self):  
        with tf.variable_scope("weights"):
           self.weights={  
                #39*39*3->36*36*20->18*18*20  
                'conv1':tf.get_variable('conv1',[11,11,3,96],initializer=tf.contrib.layers.xavier_initializer_conv2d()),  
                #18*18*20->16*16*40->8*8*40  
                'conv2':tf.get_variable('conv2',[5,5,96,256],initializer=tf.contrib.layers.xavier_initializer_conv2d()),  
                #8*8*40->6*6*60->3*3*60  
                'conv3':tf.get_variable('conv3',[3,3,256,384],initializer=tf.contrib.layers.xavier_initializer_conv2d()),  
                #3*3*60->120  
                'conv4':tf.get_variable('conv4',[3,3,384,384],initializer=tf.contrib.layers.xavier_initializer_conv2d()),

'conv5':tf.get_variable('conv5',[3,3,384,256],initializer=tf.contrib.layers.xavier_initializer_conv2d()),

'fc1':tf.get_variable('fc1',[6*6*256,4096],initializer=tf.contrib.layers.xavier_initializer()),  
                'fc2':tf.get_variable('fc2',[4096,4096],initializer=tf.contrib.layers.xavier_initializer()),

#120->6  
                'fc3':tf.get_variable('fc3',[4096,2],initializer=tf.contrib.layers.xavier_initializer()),  
                }  
        with tf.variable_scope("biases"):  
            self.biases={  
                'conv1':tf.get_variable('conv1',[96,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),  
                'conv2':tf.get_variable('conv2',[256,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),  
                'conv3':tf.get_variable('conv3',[384,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),  
                'conv4':tf.get_variable('conv4',[384,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),  
                'conv5':tf.get_variable('conv5',[256,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),  
                
                'fc1':tf.get_variable('fc1',[4096,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),  
                'fc2':tf.get_variable('fc2',[4096,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32)),
                'fc3':tf.get_variable('fc3',[2,],initializer=tf.constant_initializer(value=0.0, dtype=tf.float32))

}
        
        
        
    def inference_test(self,images):  
                # 向量转为矩阵  
        images = tf.reshape(images, shape=[-1, 39,39, 3])# [batch, in_height, in_width, in_channels]  
        images=(tf.cast(images,tf.float32)/255.-0.5)*2#归一化处理  
 
 
 
        #第一层  
        conv1=tf.nn.bias_add(tf.nn.conv2d(images, self.weights['conv1'], strides=[1, 1, 1, 1], padding='VALID'),  
                             self.biases['conv1'])  
 
        relu1= tf.nn.relu(conv1)  
        pool1=tf.nn.max_pool(relu1, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='VALID')  
 
 
        #第二层  
        conv2=tf.nn.bias_add(tf.nn.conv2d(pool1, self.weights['conv2'], strides=[1, 1, 1, 1], padding='VALID'),  
                             self.biases['conv2'])  
        relu2= tf.nn.relu(conv2)  
        pool2=tf.nn.max_pool(relu2, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='VALID')  
 
 
        # 第三层  
        conv3=tf.nn.bias_add(tf.nn.conv2d(pool2, self.weights['conv3'], strides=[1, 1, 1, 1], padding='VALID'),  
                             self.biases['conv3'])  
        relu3= tf.nn.relu(conv3)  
        pool3=tf.nn.max_pool(relu3, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='VALID')  
 
 
        # 全连接层1,先把特征图转为向量  
        flatten = tf.reshape(pool3, [-1, self.weights['fc1'].get_shape().as_list()[0]])  
 
        fc1=tf.matmul(flatten, self.weights['fc1'])+self.biases['fc1']  
        fc_relu1=tf.nn.relu(fc1)  
 
        fc2=tf.matmul(fc_relu1, self.weights['fc2'])+self.biases['fc2']  
 
        return  fc2  
 
    #计算softmax交叉熵损失函数  
    def sorfmax_loss(self,predicts,labels):  
        predicts=tf.nn.softmax(predicts)  
        labels=tf.one_hot(labels,self.weights['fc3'].get_shape().as_list()[1])  
        loss = tf.nn.softmax_cross_entropy_with_logits(predicts, labels)
      #  loss =-tf.reduce_mean(labels * tf.log(predicts))# tf.nn.softmax_cross_entropy_with_logits(predicts, labels)  
        self.cost= loss  
        return self.cost  
    #梯度下降  
    def optimer(self,loss,lr=0.01):  
        train_optimizer = tf.train.GradientDescentOptimizer(lr).minimize(loss)  
 
        return train_optimizer  
 
 
def train():  
  #  encode_to_tfrecords("data/train.txt","data",'train.tfrecords',(45,45))  
  #  image,label=decode_from_tfrecords('data/train.tfrecords')
  #  image,label=read_and_decode("/home/zenggq/data/imagedata/data.tfrecords")
    batch_image,batch_label=read_and_decode("/home/zenggq/data/imagedata/data.tfrecords")
 #   batch_image = tf.random_crop(batch_image1, [39, 39, 3])
 #   batch_image,batch_label=get_batch(image,label,batch_size=5,crop_size=227)#batch 生成测试  
   #网络链接,训练所用  
    net=network()  
    inf=net.inference(batch_image)  
    loss=net.sorfmax_loss(inf,batch_label)  
    opti=net.optimer(loss)  
    #验证集所用  
    """ encode_to_tfrecords("data/val.txt","data",'val.tfrecords',(45,45))  
    test_image,test_label=decode_from_tfrecords('data/val.tfrecords',num_epoch=None)  
    test_images,test_labels=get_test_batch(test_image,test_label,batch_size=120,crop_size=39)#batch 生成测试  
    test_inf=net.inference_test(test_images)  
    correct_prediction = tf.equal(tf.cast(tf.argmax(test_inf,1),tf.int32), test_labels)  
    accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))  """
    init=tf.initialize_all_variables()
    with tf.Session() as session:  
        with tf.device("/gpu:1"):
            session.run(init)  
            coord = tf.train.Coordinator()  
            threads = tf.train.start_queue_runners(coord=coord)  
            max_iter=9000  
            iter=0  
            if os.path.exists(os.path.join("model",'model.ckpt')) is True:  
                tf.train.Saver(max_to_keep=None).restore(session, os.path.join("model",'model.ckpt'))  
            while iter<max_iter:  
                loss_np,_,label_np,image_np,inf_np=session.run([loss,opti,batch_image,batch_label,inf])  
                #print image_np.shape  
                #cv2.imshow(str(label_np[0]),image_np[0])  
                #print label_np[0]  
                #cv2.waitKey()  
                #print label_np  
                if iter%50==0:  
                    print 'trainloss:',loss_np  
         #       if iter%500==0:  
         #           accuracy_np=session.run([accuracy])  
          #          print '***************test accruacy:',accuracy_np,'*******************'  
         #           tf.train.Saver(max_to_keep=None).save(session, os.path.join('model','model.ckpt'))  
                iter+=1  
            coord.request_stop()#queue需要关闭,否则报错  
            coord.join(threads)
       #     session.close()

if __name__ == '__main__':
    train()

tensorflow实战系列(三)一个完整的例子的更多相关文章

  1. tensorflow实战系列(一)

    最近开始整理一下tensorflow,准备出一个tensorflow实战系列,以飨读者. 学习一个深度学习框架,一般遵循这样的思路:数据如何读取,如如何从图片和标签数据中读出成tensorflow可以 ...

  2. WCF开发实战系列三:自运行WCF服务

    WCF开发实战系列三:自运行WCF服务 (原创:灰灰虫的家 http://hi.baidu.com/grayworm)上一篇文章中我们建立了一个WCF服务站点,为WCF服务库运行提供WEB支持,我们把 ...

  3. 《Java从入门到失业》第四章:类和对象(4.3):一个完整的例子带你深入类和对象

    4.3一个完整的例子带你深入类和对象 到此为止,我们基本掌握了类和对象的基础知识,并且还学会了String类的基本使用,下面我想用一个实际的小例子,逐步来讨论类和对象的一些其他知识点. 4.3.1需求 ...

  4. ElasticSearch实战系列三: ElasticSearch的JAVA API使用教程

    前言 在上一篇中介绍了ElasticSearch实战系列二: ElasticSearch的DSL语句使用教程---图文详解,本篇文章就来讲解下 ElasticSearch 6.x官方Java API的 ...

  5. 扩展Python模块系列(二)----一个简单的例子

    本节使用一个简单的例子引出Python C/C++ API的详细使用方法.针对的是CPython的解释器. 目标:创建一个Python内建模块test,提供一个功能函数distance, 计算空间中两 ...

  6. Tensorflow实战系列之五:

    打算写实例分割的实战,类似mask-rcnn. Tensorflow实战先写五个系列吧,后面新的技术再添加~~

  7. MP实战系列(三)之实体类讲解

    首先说一句,mybatis plus实在太好用了! mybaits plus的实体类: 以我博客的用户类作为讲解 package com.blog.entity; import com.baomido ...

  8. tensorflow实战系列(四)基于TensorFlow构建AlexNet代码解析

    整体流程介绍: 我们从main函数走,在train函数中,首先new了一个network;然后初始化后开始训练,训练时设定设备和迭代的次数,训练完后关闭流程图. 下面看network这个类,这个类有许 ...

  9. tensorflow实战系列(二)TFRecordReader

    前面写了TFRecordWriter的生成.这次写TFRecordReader. 代码附上: def read_and_decode(filename):    #根据文件名生成一个队列    fil ...

随机推荐

  1. Jmeter(十)检查点

    检查点又名断言,我们在手工测试过程中肉眼以及自己的逻辑思维对实际结果进行判断是否与预期结果一致,但是工具是死的,没有眼睛,没有思维,并不知道需要判断的信息在哪块,或者是来判断什么东西,我们需要让工具更 ...

  2. Mybatis 系列9-强大的动态sql 语句

    [Mybatis 系列10-结合源码解析mybatis 执行流程] [Mybatis 系列9-强大的动态sql 语句] [Mybatis 系列8-结合源码解析select.resultMap的用法] ...

  3. locals()和globals()

    都是获取当前作用域的内容: locals() 获取局部作用域的所有内容 函数内:获取locals()之前的,当前作用阈所有内容 函数外:获取打印前, 当前的作用域所有内容 在闭包内: 会把使用到的外层 ...

  4. BCGcontrolBar(六) RibbonBar编辑

    BCGcontrolBar 可以使用 Ribbon Designer方便的对 Ribbon条进行编辑 文件位置为 C:\Program Files\BCGSoft\BCGControlBarPro\D ...

  5. 解决最近vs2017总是在加载工程的时候卡住不不动

    就像这样 解决方法 删除工程目录下的.vs目录

  6. Html5——canvas标签使用

    canvas 拥有多种绘制路径.矩形.圆形.字符以及添加图像的方法. canvas 元素本身是没有绘图能力的.所有的绘制工作必须在 JavaScript 内部完成 <script type=&q ...

  7. c语言数据类型(二)

    char 类型 1.char 变量 常量 char c; 定义一个char变量 c = ‘a’   'a'字符常量 char 的本质就是一个整数,只有一个字节大小的整数 2.printf 输出char ...

  8. 04 bash程序的基本特性

    我们知道Linux系统有两种操作方式,一种为GUI的图形界面化管理方式,其中图形程序常见的有Gnome.KDE.xfce:另一种管理方式就是就是GLI的命令行管理方式,而命令行的管理方式就是通过she ...

  9. Hadoop的RPC工作原理

    RPC远程过程调用: Hadoop的远程过程调用(Remote Procedure Call,RPC)是Hadoop中核心通信机制,RPC主要通过所有Hadoop的组件元数据交换,如MapReduce ...

  10. flask表单标签

    环境:python2.7+flask+flask_wtf 内容:flask的部分常用表单标签 test.py # -*- coding: utf-8 -*- # flask web_form 表单控件 ...