Tensorflow使用杂记
零壹博弈
渐渐地,我们也会成为别人口中的大佬
展开
-
Tensorflow2.0中利用keras读取保存的模型时, 产生错误 [ValueError: Unknown activation function:relu6]
在使用已经训练好的mobilenet时候,keras产生错误 [ValueError: Unknown activation function:relu6]目前博主查到了两种解决办法:1、查看在建立模型过程中,定义 relu6 激活函数时是否使用了 tf.keras.layers.Activation(tf.nn.relu6),如果有的话,将其更改为:(记住,是所有的,更换前可以按 ctrl + F 键搜索一下当前的代码中是否还有未替换的)tf.keras.layers.ReLU(6.)2、如果已原创 2020-10-08 20:46:51 · 1502 阅读 · 1 评论 -
解决tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.
使用tensorflow2.x 时报错:tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[node model/conv2d/Con原创 2020-10-08 20:34:35 · 1037 阅读 · 1 评论 -
深度学习系列笔记——叁 (提取出神经网络的某一层特征,可视化显示)
神经网络在训练过程中很像一个黑盒,除了输入层,输出层的结果可以一目了然,中间的隐层在训练中的变化,我们不太方便查看,本篇博文将利用keras的API,对已训练好的模型在预测过程中的变化进行提取。import matplotlib.pyplot as pltimport tensorflow as tfimport numpy as npimport timeconfig = tf.compat.v1.ConfigProto(allow_soft_placement=True)config.gp原创 2020-09-12 10:25:22 · 2934 阅读 · 3 评论 -
深度学习系列笔记——壹(深度学习环境的搭建及填坑之旅,基于windows,增强版填坑)
2020.08.03补充 这一篇博文是对上一篇在windows下安装tensorflow2.1的补充,同时也包含了对于tensorflow1.15安装的注意事项。注意,此篇文章适用于Tensor flow1.15及2.1版本的安装及报错修复。由于最近项目需要TF1.x版本的,但是由于电脑安装的是TF2.1版本的,于是需要降级,然而不幸的是,电脑安装软件过多,环境混乱,于是我又利用win10的还原,清空了C盘的文件。接下来就是各种踩坑,再次记录一下,方便后来者:首先,我这次需要安装的是TF1.15,原创 2020-08-29 17:31:22 · 235 阅读 · 0 评论 -
深度学习系列笔记——贰 (基于Tensorflow2 Keras迁移学习,使用预训练模型解决猫狗大战 四)
深度学习系列笔记——贰 (基于Tensorflow Keras搭建的猫狗大战模型 一)深度学习系列笔记——贰 (基于Tensorflow Keras搭建的猫狗大战模型 二)深度学习系列笔记——贰 (基于Tensorflow2 Keras搭建的猫狗大战模型 三)本篇博客是对于之前猫狗大战的补充,猫狗大战使用的是我们自己的模型,输入的图像尺寸是200x200x3,接下来,在之前的基础上,我们更改一下模型,并且把之前的输入尺寸进行一定的改动,使用经典的VGGNet(包含vgg16和vgg19),提升我们模型原创 2020-08-27 21:12:50 · 1202 阅读 · 2 评论 -
深度学习系列笔记——杂记(TensorBoard使用)
博主使用的是TF2.1版本所用的Tensorboard ,代码结构如下#这里设置的是回调,用于断点续训,详情可百度之cp_callback = tf.keras.callbacks.ModelCheckpoint( filepath=checkpoint_save_path, save_weights_only=True, save_best_only=True)tb_callbacks = tf.keras.callbacks.TensorBoard( log_dir=原创 2020-08-23 10:51:29 · 458 阅读 · 0 评论