TensorFlow
文章平均质量分 67
KyrieLiu52
这个作者很懒,什么都没留下…
展开
-
【Qt】控制台信息输出到GUI中(在TensorFlow、PySide6环境下,将训练信息实时输出在文本框中)
1.前言本文主要是介绍如何将控制台的内容实时的输出到GUI中。主要实现方式就是通过多线程,一个GUI的线程,还有一个线程提供给想要输出的程序。需要两个线程的原因是,GUI主线程一旦启动,直到下一次调用刷新页面功能的操作出现之前,都无法改变当前界面,因此需要另外一个线程来改变GUI的内容。2.实现实现功能为:将训练模型的信息,输出到GUI中的文本框中,假设文本框为 W。思想如下:在GUI中,定义一个如何更新 W 中内容的函数 onUpdateText()自定义一个输出流 Stream ,将某原创 2021-09-13 17:31:05 · 3820 阅读 · 5 评论 -
【TensorFlow】神经网络中间层截取、可视化中间层结果
在预测的过程中,想要将神经网络模型的中间层结果获取到,并进行可视化。训练过程中搭建的模型代码如下:class_num = 3base_model = tf.keras.applications.MobileNetV2(input_shape=IMG_SHAPE, include_top=False, weights=原创 2021-09-13 16:37:10 · 3213 阅读 · 0 评论 -
【TensorFlow】如何通过手动方式下载TensorFlow官方的ImageNet预训练模型
在使用TensorFlow进行网络的搭建时,可以使用下面的语句来获取一些网络的模型(其中:include_top表示是否包括顶层网络,顶层即处理输出的几个层;weights表示模型加载的权重,可以选择None不加载、imagenet加载网络在ImageNet上训练的权重、或者指定本地权重的路径)base_model = tf.keras.applications.MobileNetV2(input_shape=IMG_SHAPE,原创 2021-08-12 15:27:04 · 3233 阅读 · 0 评论 -
MVSNet代码复现问题解决、cuBlas call failed status=13问题
简介这篇文章记录复现MVSNet中遇到的一些问题。最近在复现 MVSNet 的代码,是一篇2018年的论文,使用的是tensorflow-gpu>=1.5的代码源码中是建议使用cuda9.0和cudnn7.0,因此根据tensorflow和cuda的版本对应,只能安装1.5~1.12版本的,最后选择安装了1.5的gpu版本tensorflow。服务器上的CUDA版本是11.2的,所以需要安装一个CUDA9.0,因为是在一起使用的服务器,所以CUDA9.0安装在个人目录下,而不是系统中,并可以在原创 2021-11-20 20:39:20 · 3415 阅读 · 6 评论 -
【TensorFlow】:GetNext() failed because the iterator has not been initialized
错误FailedPreconditionError (see above for traceback): GetNext() failed because the iterator has not been initialized. Ensure that you have run the initializer operation for this iterator before getting the next element. [[Node: IteratorGetNext = IteratorGe原创 2021-11-09 17:36:48 · 2419 阅读 · 0 评论