Tensorflow
文章平均质量分 79
JinSu_
这个作者很懒,什么都没留下…
展开
-
【Tensorflow】keras模型预测+keras模型转tflite模型(post-training模型量化)+tflite模型预测+对比量化模型和原始模型的推理速度和文件大小
本文python程序在在tensorflow2.4.1上运行,CUDA11.0,cuDNN8.0.5之前已经在keras上用多种数据输入方式训练得到keras模型【Tensorflow】训练keras模型+keras的数据生成器ImageDataGenerator+jpg图像数据格式的MNIST数据集+对比flow和flow_from_directory【Tensorflow】训练keras模型+tensorflow-V2的数据集tf.data.Dataset+jpg图像.原创 2021-02-04 11:13:05 · 3539 阅读 · 0 评论 -
【Tensorflow】简明记录tf.data.Dataset类的成员方法
tf.data.Dataset的简单介绍https://tensorflow.google.cn/versions/r2.1/api_docs/python/tf/data/Datasettf.data.Dataset API支持编写描述性和高效的输入管道。tf.data.Dataset的使用遵循一个常见的模式:从输入数据创建源数据集。 应用数据集转换方法来预处理数据。 迭代数据集并处理元素。迭代以流方式进行,因此不需要将整个数据集装入内存。tf.data.Da...原创 2021-02-03 15:45:17 · 321 阅读 · 0 评论 -
【Tensorflow】tf.image的部分方法详解+基于随机子空间方法(RSM)的图像增强
基于随机子空间方法(RSM)的图像增强在数据图像处理中,Random Erasing是指随机选择图像中的一个或者多个区域进行擦除的操作,擦除之后的图像是原图像的一个随机子空间RSM,其随机保留了图像样本的部分特征而不是全部特征。若结合集成学习的方法训练多个网络,可以在不进行数据增强的情况下增加训练样本的数量,并且提高模型的泛化能力。import matplotlib.pyplot as pltimport tensorflow as tfimport numpy as npimport原创 2021-02-03 15:02:09 · 911 阅读 · 0 评论 -
【Tensorflow】tensorflow-V1版本+jpg图像数据格式的MNIST数据集+tf.train.batch批量数据生成+静态graph的建立和训练过程
谨以此文纪念逝去的青春,再见了tensorflow-V1!!!input_data.pyimport tensorflow as tfimport numpy as npimport os#%%# you need to change this to your data directory#train_dir = 'E:/MNISTdata/'def get_files(file_dir): ''' Args: file_dir: file.原创 2021-01-31 17:02:40 · 379 阅读 · 0 评论 -
【Tensorflow】训练keras模型+keras的数据生成器ImageDataGenerator+jpg图像数据格式的MNIST数据集+对比flow和flow_from_directory
1.数据集jpg图像数据格式的MNIST数据集:(放在database1文件夹下面)2.利用tkeras的数据生成器ImageDataGenerator去训练keras模型(MobileNet)通过tf.io.file读取图像文件,然后用tf.image.decode_jpeg转文件格式,保存为np.array(list(float,int))格式这样子的话,图像数据集就是全部被读取到内存中,这是非常占内存的!!!接下来可以有两种方法训练keras模型:np.array(.原创 2021-01-30 20:49:23 · 1255 阅读 · 0 评论 -
【Tensorflow】训练keras模型+tensorflow-V2的数据集tf.data.Dataset+jpg图像数据格式的MNIST数据集+MobileNet
训练过程如下:利用pathlib和python自带的list,将原始图像转为list(string,int)格式利用tf.data.Dataset.from_tensor_slices将list数据转为tf.data.Dataset中的BatchDataset,然后用tf.data.Dataset.zip合并两个BatchDataset为一个ZipDataset构建keras网络模型:from tensorflow_core.python.keras.applications.mobilenet i原创 2021-01-30 08:12:36 · 1508 阅读 · 1 评论 -
【Tensorflow】tensorflow和keras+读取jpg图像数据格式的MNIST数据集
1.数据集介绍jpg图像数据格式的MNIST数据集:(放在database1文件夹下面)2.tensorflow读取jpg图像数据格式的mnist数据集:tensorflow1.x的读取方式:tensorflow1.12以上的读取方式:(最好是1.13.1或者2.x)https://blog.csdn.net/Black_Friend/article/details/104529859import tensorflow as tfimport rand...原创 2021-01-29 21:08:50 · 723 阅读 · 0 评论 -
【Tensorflow】tensorflow和keras读取官方版本的MNIST数据集
1.数据集介绍tensorflow官方mnist:(放在database2文件夹下面)keras官方mnist:(放在database3文件夹下面)jpg图像数据格式的MNIST数据集:(放在database文件夹下面)2.tensorflow读取tensorflow官方mnist:import tensorflow as tf import tensorflow.examples.tutorials.mnist.input_data as input_...原创 2021-01-29 20:10:04 · 3707 阅读 · 0 评论 -
【Tensorflow】GPU版本的tensorflow的系统环境配置要求
tensorflow安装前需要清楚,tensorflow的软件要求,GPU驱动、CUDA、cuDNN的版本要求!tensorflow中文网:https://tensorflow.google.cn/install/gpu原创 2021-01-27 20:58:03 · 7960 阅读 · 2 评论 -
【Tensorflow】tensorflow预训练简单模型及权重文件复用初始化复杂模型
笔者在学习YOLO网络的过程中,遇到了预训练问题。我在网上搜到的大部分相关问题都是在说如何利用之前的预训练权重文件做fineturning问题:如何预训练简单网络,然后复用权重文件初始化复杂网络?在yolo中,我需要预训练前20层网络,使其能在物体分类上达到不错的准确率然后复用这个简单网络的权重文件初始化正式的YOLO网络,这是我遇到的实际问题一开始,我想问题的突破口有三...原创 2017-08-11 21:12:50 · 6830 阅读 · 0 评论 -
【 Tensorflow】Tensorflow1.x的一些学习体会
虽然Tensorflow2.0已经发布了!但是我还是想写一些使用tensorflow1.x这一路的一些心得体会反正2.0也是会兼容1.x的,哈哈哈(自我安慰(怎么就没有时间跟上google大佬们的步伐呢?))简单来说2.0最大的亮点就推出Eager Execution模式,以便于模型的快速迭代开发。但依然包含传统的Graph Execution模式在1.x的时候,我们是要建立gra...原创 2019-11-27 19:14:26 · 544 阅读 · 0 评论 -
【Tensorflow】toco编译+利用toco进行pb转tflite+模型量化
编译tensorflow里面的toco工具:(bazel安装参考:https://blog.csdn.net/jin739738709/article/details/86705735?spm=1001.2014.3001.5501)cd /d D:\lib\tensorflow\tensorflow\lite\toco (去到\tensorflow-master\tensorflow\lite\toco)bazel build :toco输入toco命令行进行pb转tflit...原创 2021-01-27 17:26:33 · 1418 阅读 · 0 评论