Fork me on GitHub

随笔分类 -  TensorFlow

摘要:TensorRT是一个用于高性能深度学习推理的SDK,它可以将已经训练好的网络转换为TensorRT引擎,从而提高推理速度和效率。TensorRT的作用有以下几点¹²: TensorRT可以对网络进行压缩、优化以及运行时部署,并且没有框架的开销。 TensorRT通过合并层、选择最优的卷积核、根据指 阅读全文
posted @ 2023-05-05 15:27 stardsd 阅读(425) 评论(0) 推荐(0) 编辑
摘要:本文主要介绍PyTorch框架下的可视化工具Tensorboard的使用 面向第一次接触可视化工具的新手<其实是备忘> 之前用了几天visdom,用起来很方便,但是画的图显得很乱,所以花了一晚上把代码里的visdom都改成了tensorboard。 Tensorboard 安装 原本是tensorf 阅读全文
posted @ 2021-03-11 11:17 stardsd 阅读(9486) 评论(0) 推荐(2) 编辑
摘要:N: batch; C: channel H: height W: width Caffe 的Blob通道顺序是:NCHW; Tensorflow的tensor通道顺序:默认是NHWC, 也支持NCHW,使用cuDNN会更快; Pytorch中tensor的通道顺序:NCHW TensorRT中的t 阅读全文
posted @ 2021-02-25 15:16 stardsd 阅读(2576) 评论(0) 推荐(0) 编辑
摘要:报错情况: 解决方案: 1.在jupyter terminal 中 进入python import tensorflow as tf 没问题 ,在ipython中执行报 ImportError: libcublas.so.10.0: cannot open shared object file: N 阅读全文
posted @ 2019-07-05 00:37 stardsd 阅读(3729) 评论(0) 推荐(0) 编辑
摘要:Deepo is a series of Docker images that allows you to quickly set up your deep learning research environment supports almost all commonly used deep le 阅读全文
posted @ 2019-07-01 20:02 stardsd 阅读(425) 评论(0) 推荐(0) 编辑
摘要:TensorFlow Docker requirements Note: To run the docker command without sudo, create the docker group and add your user. For details, see the post-inst 阅读全文
posted @ 2019-07-01 19:51 stardsd 阅读(407) 评论(0) 推荐(0) 编辑
摘要:版权声明:本文为博主原创文章,如需转载请注明出处,谢谢。 https://blog.csdn.net/qq_38542085/article/details/78742295 初始代码 import pandas as pd import numpy as np import tensorflow 阅读全文
posted @ 2019-03-19 11:51 stardsd 阅读(1998) 评论(0) 推荐(0) 编辑
摘要:import tensorflow 报错: tf.estimator package not installed. 解决方案1: 安装 解决方案2: 阅读全文
posted @ 2019-03-15 18:26 stardsd 阅读(2687) 评论(0) 推荐(0) 编辑
摘要:nvidia-smi命令输出如下: 解释: 第一栏的Fan:N/A是风扇转速,从0到100%之间变动,这个速度是计算机期望的风扇转速,实际情况下如果风扇堵转,可能打不到显示的转速。有的设备不会返回转速,因为它不依赖风扇冷却而是通过其他外设保持低温(比如我们实验室的服务器是常年放在空调房间里的)。 第 阅读全文
posted @ 2019-03-06 14:39 stardsd 阅读(8117) 评论(0) 推荐(0) 编辑
摘要:要运行高版本的GPU版TensorFlow,需要更新宿主机的显卡驱动(本文以NVIDIA390为例) 一、更新驱动 卸载已有驱动 sudo apt-get remove --purge nvidia* 二、报错: 说是驱动有问题,第一个想到的就是因为将早先的 nvidia-384 驱动更新到了 nv 阅读全文
posted @ 2019-03-02 21:36 stardsd 阅读(17625) 评论(0) 推荐(0) 编辑
摘要:Keras有两种类型的模型,序贯模型(Sequential)和函数式模型(Model),函数式模型应用更为广泛,序贯模型是函数式模型的一种特殊情况。 两类模型有一些方法是相同的: model.summary():打印出模型概况,它实际调用的是keras.utils.print_summary mod 阅读全文
posted @ 2019-01-29 21:55 stardsd 阅读(4600) 评论(1) 推荐(0) 编辑
摘要:七 测试网络 模型测试包含于test.py文件,Detector类的image_detector()函数用于检测目标。 1、类初始化函数 2、draw_result()函数 在原始图像上绘制边界框,并添加一些附件信息,如目标类别,置信度。 3、detect()函数 detect()函数用来对图像进行 阅读全文
posted @ 2019-01-18 15:58 stardsd 阅读(1715) 评论(0) 推荐(0) 编辑
摘要:五 读取数据pascal_voc.py文件解析 我们在YOLENet类中定义了两个占位符,一个是输入图片占位符,一个是图片对应的标签占位符,如下: 而pascal_voc.py文件的目的就是为了准备数据,赋值给占位符。在pascal_voc.py文件中定义了一个pascal_voc,该类包含了类初始 阅读全文
posted @ 2019-01-18 15:57 stardsd 阅读(1625) 评论(0) 推荐(0) 编辑
摘要:tf.transpose函数中文意思是转置,对于低维度的转置问题,很简单,不想讨论,直接转置就好(大家看下面文档,一看就懂)。 本文主要讨论高维度的情况: 为了形象理解高维情况,这里以矩阵组合举例: 先定义下: 2 x (3*4)表示2个3*4的矩阵,(其实,它是个3维张量)。 x = [[[1,2 阅读全文
posted @ 2019-01-18 15:43 stardsd 阅读(12898) 评论(0) 推荐(0) 编辑
摘要:主要通过whl方式进行配置。 1.1 Whl文件下载地址: https://www.lfd.uci.edu/~gohlke/pythonlibs/ 注意:必须安装numpy-mkl,替代原有的numpy,安装时会自动卸载原版本。 共下载三个文件: numpy-1.16.0+mkl-cp37-cp37 阅读全文
posted @ 2019-01-18 10:04 stardsd 阅读(32920) 评论(0) 推荐(0) 编辑
摘要:前言在跑通了官网的mnist和cifar10数据之后,笔者尝试着制作自己的数据集,并保存,读入,显示。 TensorFlow可以支持cifar10的数据格式, 也提供了标准的TFRecord 格式,而关于 tensorflow 读取数据, 官网提供了3中方法 1 Feeding: 在tensorfl 阅读全文
posted @ 2018-11-04 15:44 stardsd 阅读(3208) 评论(0) 推荐(0) 编辑
摘要:问题描述 这几天在用TensorFlow搭建一个神经网络来做一个binary classifier,搭建一个典型的神经网络的基本思路是: 定义神经网络的layers(层)以及初始化每一层的参数 然后迭代: 前向传播(Forward propagation) 计算cost(Compute cost) 阅读全文
posted @ 2018-03-08 10:34 stardsd 阅读(5221) 评论(0) 推荐(0) 编辑
摘要:搭建普通的卷积CNN网络。 nan表示的是无穷或者是非数值,比如说你在tensorflow中使用一个数除以0,那么得到的结果就是nan。 在一个matrix中,如果其中的值都为nan很有可能是因为采用的cost function不合理导致的。 当使用tensorflow构建一个最简单的神经网络的时候 阅读全文
posted @ 2018-03-07 23:46 stardsd 阅读(4154) 评论(0) 推荐(0) 编辑
摘要:Returns the index with the largest value across axis of a tensor. input is a Tensor and axis describes which axis of the input Tensor to reduce across 阅读全文
posted @ 2018-03-07 21:16 stardsd 阅读(1196) 评论(0) 推荐(0) 编辑