TensorFlow 1.X
TensorFlow 1.X学习
太阳照常升起!
这个作者很懒,什么都没留下…
展开
-
图像转成tfrecords很大的问题
tfrecords文件过大问题使用默认方式将图像转成tfrecords文件很大使用默认方式将图像转成tfrecords文件很大解决方法:构造writer时,加上options参数——options = tf.python_io.TFRecordOptions(compression_type=tf.python_io.TFRecordCompressionType.ZLIB)writer = tf.python_io.TFRecordWriter(tfrecords_path, options)原创 2021-01-11 17:06:31 · 386 阅读 · 0 评论 -
TensorFlow各个版本对应的Python,CUDA,和cuDNN
参考TensorFlow官方文档TF官方文档中文原创 2021-01-06 11:30:25 · 900 阅读 · 0 评论 -
TensorFlow将meta转化为summary,从ckpt查看tensor的name
meta转化为summary用tensorboard查看import tensorflow as tffrom tensorflow.python.platform import gfilegraph = tf.get_default_graph()graphdef = graph.as_graph_def()_ = tf.train.import_meta_graph('meta的路...原创 2019-12-25 21:51:30 · 345 阅读 · 0 评论 -
TensorFlow slim代码示例
slim代码示例,包括读取tfrecords,带BN的训练,L2正则化读取tfrecordsimport globimport cv2import numpy as npimport tensorflow as tf# 从自定义的config.py里导入一些参数from config import HEIGHT, WIDTH, CHANNELdef read(file_queue...原创 2019-12-25 21:37:39 · 263 阅读 · 0 评论 -
TensorFlow用GPU训练
TensorFlow用GPU训练#配置GPUconfig = tf.ConfigProto()config.gpu_options.per_process_gpu_memory_fraction = 0.7 # 显存占用率config.allow_soft_placement = Trueconfig.log_device_placement = Trueconfig.gpu_opt...原创 2018-11-24 10:44:06 · 2302 阅读 · 0 评论 -
TensorFlow模型变量重用
TensorFlow模型变量重用问题加载模型及检测的.py# predict.pyimport numpy as npimport tensorflow as tffrom PIL import Imageimport timeimport fcrndef predict(model_data_path, image_path , png_path): # Defau...原创 2018-11-28 12:04:52 · 897 阅读 · 1 评论 -
TensorFlow slim中正则化的使用
TensorFlow slim中正则化的使用关于slim中的正则化使用,以L2正则化为例:# 构建网络时候添加l2正则化with slim.arg_scope([slim.conv2d], weights_regularizer=slim.l2_regularizer(0.0001)): net = slim.conv2d(inputs, 64, 5, 2, padding='SAME',...原创 2019-04-05 21:54:08 · 435 阅读 · 0 评论 -
TensorFlow实现batch_borm
TensorFlow实现batch_bormdef batch_norm(x, is_training, eps=EPS, decay=0.9, affine=True, name='batch_norm'): from tensorflow.python.training.moving_averages import assign_moving_average ...原创 2019-04-28 16:36:51 · 194 阅读 · 0 评论 -
TensorFlow实现反池化
TensorFlow实现2×2反池化2×2反池化如下图,将输入的尺寸扩大为原来两倍,输入值填充到新的每个2×2网格的左上角,其余三个填0。代码实现TensorFlow中没有反池化函数,以下是代码实现。# 2x2反池化def unpool(value, name='unpool'): with tf.name_scope(name) as scope: sh =...原创 2019-04-29 16:28:51 · 1373 阅读 · 2 评论