tensorflow学习
星辰辰大海
这个作者很懒,什么都没留下…
展开
-
新装的TensorFlow出现protobuf相关错误
今天刚装了个TensorFlow2.4,import tensorflow的时候,出现如下错误TypeError: Descriptors cannot not be created directly.If this call came from a _pb2.py file, your generated code is out of date and must be regenerated with protoc >= 3.19.0.If you cannot immediately re原创 2022-05-30 16:51:59 · 16911 阅读 · 20 评论 -
解决tensorflow2.4.0版本出现failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED错误的方法
方法参考了https://github.com/tensorflow/tensorflow/issues/45070在主代码中加入physical_devices = tf.config.list_physical_devices('GPU')tf.config.experimental.set_memory_growth(physical_devices[0], True)即可解决这个错误相当简单...原创 2021-02-26 09:38:01 · 2100 阅读 · 5 评论 -
从tf.keras模型的中间层开始输入的方法
目的想要分割一个模型,直接使用 tf.keras.Model(inputs=……, outputs=……)方法只能从模型本身设定的输入作为新模型的输入。如果想要从中间层开始输入,封装一个新的模型需要比较麻烦的操作。实现原理手动设定一个输入,然后函数式的调用每一层的输入,最后调用 tf.keras.Model(inputs=……, outputs=……)封装模型即可例子我使用的是tensorflow2.4。以不带SE模块的mobilenetv3为例,我们需要保存每一层的输出和他们的名称原创 2021-02-20 16:30:35 · 1717 阅读 · 1 评论 -
简单demo带你将tensorflow2.x的自定义keras模型转为tflite格式并部署到安卓端
环境:windows 10CUDA 10.1cudnn 7.6.4tensorflow-gpu 2.1androidstudio 3.6基本都是目前比较新的环境。因为tensorflow2.0后都喜欢使用keras自定义模型,所以部署方法与原来的有些不同,但是过程还是相当简单的。直接进入主题。首先是python部分:1.自定义一个多输入多输出的简...原创 2020-02-25 16:34:29 · 2049 阅读 · 4 评论 -
简单的demo带你将TensorFlow的pb格式的模型移植到Android平台
最近想把在tensorflow上训练的模型移植到安卓上运行,看了一下网上的例子,感觉都很复杂,对于我这种不太会安卓代码的人很不友好,所以决定自己摸索,再看了tensorflow官方的demo后,决定写出了下面这个简易版demo,带你快速了解如何将pb模型移植到安卓上运行。我的环境:windows10 python3.7 tensorflow-gpu1.14 pycharm andr...原创 2019-07-26 11:59:10 · 4043 阅读 · 14 评论