tensorflow
关注
分享
扫一扫
tensorflow相关学习总结
galaxyzwj
Code the Future, change the world
展开
-
tensorflow2------猫狗数据集的分类以及模型部署运行在jetson nano上
1. 猫狗图片数据集下载首先我们需要做一些准备工作，即准备需要训练模型的猫狗图片数据集。这里我们猫狗数据集图片来源于kaggle中Dogs vs. Cats数据集，下载后我们提取了其中的猫的图片和狗的图片各1500张，训练集中帽钩图片各1000张，验证集中猫狗图片各500张。并新建了一个test文件夹，用于存放我们需要预测的一些猫狗图片资源。文件夹目录下： test train validation2. 迁移学习resnet50并进行模型训练这里我们使用迁移学习使用resnet50的模型..原创 2020-05-09 18:15:26 · 2482 阅读 · 0 评论 -
jetson nano上部署运行 tensorflow lite模型
在移动端这里就不使用python而是使用C++作为开发语言，总体流程就是我们在PC端交叉编译出相关执行程序，然后在jetson nano上直接运行，不考虑在jetson nano上编译，使用交叉编译这也更加符合嵌入式软件开发的流程(虽然jetson nano和raspberry pi都自带系统可以下载编译代码。。。。。)1. PC机上交叉编译出tflite静态库首先需要做的是在PC机上从g...原创 2020-05-07 18:29:44 · 2704 阅读 · 1 评论 -
tensorflow2------保存加载模型并转换为TF Lite
1. callbacks.ModelCheckpoint中保存keras h5模型、权重参数save_weights_only=Trueoutput_model_file=os.path.join(logdir,"fashion_mnist_weights.h5")#在logdir中创建一个模型文件.h5#定义一个callbacks数组callbacks = [ keras...原创 2020-05-08 23:01:21 · 4733 阅读 · 0 评论 -
tensorflow2------cifar-10图像分类问题
cifar-10数据集来自于kaggle平台上下载下来的(一般新推出的模型论文都会使用cifar数据集，但是使用的是cifar-100,是针对100类不同对象的分类)，我们通过在本地建立模型来实现该10分类问题，并将test数据集上的识别结果生成csv文件上传到kaggle上进行验证，可以得到识别test数据的分数。这里代码使用的是基本的卷积神经网络模型(conv + bn + max_poo...原创 2020-05-04 21:58:42 · 1282 阅读 · 0 评论 -
tensorflow2------10-monkey && ResNet50 &&迁移学习
10-monkey-species 数据集是一个10类不同品种猴子的数据集，这个数据集是从kaggle平台中下载到本地使用的，在这个分类猴子的数据集中我们使用resnet50模型来做迁移学习fine tune，并且最终实现向模型中输入一张图片能够打印出该图片属于哪类猴子品种的结果。import matplotlib as mpl #画图用的库import matplotlib.pyplot...原创 2020-05-02 00:17:42 · 992 阅读 · 0 评论 -
tensorflow2------深度可分离卷积的实现
深度可分离卷积(Depthwise separable convolution)是原创 2020-04-28 23:38:09 · 2385 阅读 · 1 评论 -
tensorflow2------cnn实现
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-27 22:50:26 · 1436 阅读 · 0 评论 -
tensorflow2------tfrecord文件的创建与读取
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-27 15:57:59 · 1384 阅读 · 0 评论 -
tensorflow2------csv文件的创建与读取
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-25 22:45:58 · 1148 阅读 · 0 评论 -
tensorflow2------dataset基础API
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-25 15:59:58 · 573 阅读 · 0 评论 -
tensorflow2------自定义求导
我们在前面的模型的训练过程中model.compile函数中指定的optimizer就是求导的算法类型，一般我们直接指定为sgd、adam算法求导，这里我们可以不依赖于tensorflow中相关算法求导，转而选择使用自定义的算法求导，这样就可以满足一些特定的需求。...原创 2020-04-24 00:50:20 · 521 阅读 · 0 评论 -
tensorflow2------tf.function 和 autograph
TensorFlow 2.0引入的eager提高了代码的简洁性，而且更容易debug。但是对于性能来说，eager执行相比Graph模式会有一定的损失。这不难理解，毕竟原生的Graph模式是先构建好静态图，然后才真正执行。这对于在分布式训练、性能优化和生产部署方面具有优势。但是好在，TensorFlow 2.0引入了tf.function和AutoGraph来缩小eager执行和Graph模式的性...原创 2020-04-22 22:13:54 · 405 阅读 · 0 评论 -
tensorflow2------自定义损失函数和Layer
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-22 16:08:13 · 966 阅读 · 0 评论 -
tensorflow2------基础API的使用
这里主要介绍了tensorflow中常量和变量的使用。常量使用的是tf.constant或tf.ragged.constant来创建变量使用的是tf.Variable来创建import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inline...原创 2020-04-21 00:56:19 · 285 阅读 · 0 评论 -
tensorflow2------超参数搜索
1.超参数超参数指的是：神经网络中有很多训练过程中不变化的参数，一般是在训练之前就已经认为设定好的，不像神经单元中权重与偏置都是在模型训练过程中值不断改变的。 网络结构参数：几层，每层宽度，每层激活函数等 训练参数：batch_size，学习率，学习率衰减算法等(batch_size:指的是一次训练所选取的样本数。在没有使用Batch Size之前，这意味着网络在训练时，是一次把...原创 2020-04-20 01:32:48 · 1114 阅读 · 0 评论 -
tensorflow2------dnn实现
在tensorflow2上实现深度神经网络的模型搭建与训练过程。这里重点说明一下搭建模型过程中有使用到 批归一化以及Dropout、selu激活函数等。import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy...原创 2020-04-17 01:39:19 · 2040 阅读 · 0 评论 -
tensorflow2------回归问题california_housing_dataset
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-16 22:46:25 · 1150 阅读 · 0 评论 -
tensorflow2------分类问题fashion_mnist
import matplotlib as mpl #画图用的库import matplotlib.pyplot as plt#下面这一句是为了可以在notebook中画图%matplotlib inlineimport numpy as npimport sklearn #机器学习算法库import pandas as pd #处理数据的库 import osimport...原创 2020-04-13 22:55:27 · 459 阅读 · 1 评论 -
PyCharm中的tensorflow配置指导
tensorflow我们根据正常的流程在ubuntu中安装好后，接下来我们想使用一个专门写python代码的IDE工具，这里我发现了一个非常好用的IDE工具-PyCharm，在这个IDE中我们可以很方便的写python代码，并且该IDE还带自动补全的功能，非常好用。前面的怎么下载、安装这个PyCharm就不写了，网上很多，我只介绍如何使用在安装好这个IDE后如何配置，使其能够在ubuntu中使...原创 2019-03-13 22:23:27 · 2416 阅读 · 0 评论 -
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
训练模型时老是报上面的错误，不断修改batch size也没有效果执行命令 watch -n 0.1 nvidia-smi 可以看到看到GPU 显存爆了。。。这里给出网上找到的一个适用于tensorflow-gpu2.0.0的方法，在前面加上如下代码即可：physical_devices = tf.config.experimental.list_physical_devic...原创 2019-12-29 00:03:21 · 687 阅读 · 0 评论 -
tensorflow 安装命令
python环境为 python-3.6.8安装cpu版本：sudo pip3 install -ihttps://pypi.tuna.tsinghua.edu.cn/simpletensorflow卸载:sudo pip3 uninstall tensorflow原创 2019-12-19 22:58:24 · 193 阅读 · 0 评论 -
tensorflor简介
tensorflow原创 2020-04-11 22:24:37 · 635 阅读 · 0 评论