Tensorflow && Keras
tf && keras
wwwsssZheRen
Zhejiang University, Computer vision
展开
-
tensorflow2.1 GPU配置
一、获得当前主机上特定运算设备的列表gpus = tf.config.experimental.list_physical_devices(device_type='GPU')cpus = tf.config.experimental.list_physical_devices(device_type='CPU')print(gpus, cpus)二、设置当前程序可见的设备范围 默认情况下 TensorFlow 会使用其所能够使用的所有 GPU。设置之后,当前程序...原创 2020-07-02 10:45:51 · 1648 阅读 · 0 评论 -
Keras实现迁移学习
第一步先用Keras的已拥有的model.load_weights(best_model_file)的方法把之前训练好的500分类的模型的权重先加载进来,现在的模型是就的500分类的模型; 第二步是在最后一层的全连接层里在原有的500分类的权重基础上,再在张量矩阵里面加上100新添加分类数的随机数矩阵。这样子就可以使用之前已经训练好的500的分类的权重,又可以在新的训练里面添加上新的分类数。 第三步把新的全连接放到Model的outputs里面创...原创 2020-05-29 14:46:16 · 802 阅读 · 0 评论 -
Tensorflow2.0版本Keras输出frozen_graph.pb格式
opencv的dnn模块中有通过readNetFromTensorflow这一个函数读入tensorflow模型,原创 2020-05-08 10:54:54 · 2162 阅读 · 0 评论 -
keras主要模块
Input(…): 用于实例化一个输入 Tensor,作为神经网络的输入。average_pooling1d(…): 一维平均池化层average_pooling2d(…): 二维平均池化层average_pooling3d(…): 三维平均池化层batch_normalization(…): 批量标准化层conv1d(…): 一维卷积层conv2d(…): 二维卷积层c...原创 2020-03-02 16:48:54 · 695 阅读 · 0 评论 -
Keras 顺序模型 && 函数式模型
Keras Sequential顺序模型是多个网络层的线性堆叠。你可以通过将网络层实例的列表传递给Sequential的构造器。以下用为mnist构建的经典网络案例。from tensorflow import kerasmodel = keras.models.Sequential([ keras.layers.Dense(32, input_shape=(7...原创 2020-03-08 10:31:38 · 714 阅读 · 0 评论 -
keras basic FAQ
https://github.com/keras-team/keras-docs-zh原创 2020-03-08 21:59:24 · 255 阅读 · 0 评论 -
tf 数值定义与计算
1、tf常量(tf.constant)1)生成全0数组# 语法tf.zeros(shape, dtype, name) # 生成两行三列的全0数组A_const = tf.zeros([2, 3], dtype='int32')2)2、变量(tf.Variable)...原创 2020-02-26 11:29:56 · 3794 阅读 · 0 评论 -
Keras回调函数Callbacks
1、tensorboard简介 tensorboard提供了可视化界面及为机器学习过程中提供所需工具。 1)跟踪并可视化像损失(loss)和精度(accuracy)这样的矩阵。 2)可视化模型图(包括ops和layer) 3)随着时间变化而变化的权重(weights)、偏置(biases)及其他tensor的直方图可视化 。。。...原创 2020-02-23 18:00:53 · 805 阅读 · 0 评论 -
Tensorflow Basic
1、tensorflow1.0 to 2.0 当安装的版本是Tensorflow2.0,若要沿用Tensorflow1.0版本中的代码的编程风格,只需要在头部增加两行代码即可。import tensorflow.compat.v1 as tftf.disable_v2_behavior()...原创 2020-02-19 14:57:23 · 367 阅读 · 0 评论 -
Tensorflow模型保存与调用
1、tensorflow模型保存为pb文件示例代码如下import tensorflow.compat.v1 as tftf.disable_v2_behavior()import osfrom tensorflow.python.framework import graph_utilpb_file_path = os.getcwd()with tf.Session(gra...原创 2020-02-20 13:59:22 · 1238 阅读 · 0 评论 -
tensorflow 数据集
1、关于tf数据集 需要加入tf.enable_eager_execution()import tensorflow.compat.v1 as tftf.disable_v2_behavior()import tensorflow_datasets as tfdsimport numpy as npimport matplotlib.pyplot as plt#...原创 2020-02-20 17:44:41 · 642 阅读 · 0 评论