![](https://img-blog.csdnimg.cn/20190927151117521.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Caffe/TF/Pytorch框架
文章平均质量分 59
深度学习框架
超级无敌陈大佬的跟班
站在巨人的肩膀上看星星.
不辜负每个朝阳,不荒废每个深夜,因平凡而奋斗,因奋斗而不平凡
展开
-
【模型部署】Tensorrt学习记录
tensorrt官方文档:https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-723一、简介TensorRT是一种深度学习框架?是一种库?官方文档中的称呼是:TensorRT 是一个 SDK(Software Development Kit)软件开发工具,用于优化经过训练的深度学习模型以实现高性能推理。加载engine模型参数"step0.通过文件流读取engine文件"std::vector<ch原创 2021-06-26 21:47:30 · 2300 阅读 · 1 评论 -
【caffe】caffe中网络层含义
Reshape层:(改变blob的形状,N,C,W,H)Reshape layer只改变输入数据的维度,但内容不变,也没有数据复制的过程,与Flatten layer类似。输出维度由reshape_param 指定,正整数直接指定维度大小,下面两个特殊的值:0 => 表示copy the respective dimension of the bottom layer,复制输入相应维度的值。 -1 => 表示infer this from the other dimension原创 2021-06-23 16:29:59 · 93 阅读 · 0 评论 -
陈大佬的pytorch使用笔记
torch.utils.data.DataLoader:pytorch的数据加载器。https://blog.csdn.net/ftimes/article/details/105201002原创 2020-08-18 14:55:25 · 182 阅读 · 0 评论 -
陈大佬的python 使用笔记
.transpose()功能:改变序列的顺序。# 改变序列的顺序,原始的为(0,1),这里使用(1,0)就是进行了转置preds = preds.transpose(1, 0).contiguous().view(-1)功能:有些tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式。#在使用view()前,需要先使用cont.原创 2020-06-28 17:23:32 · 304 阅读 · 0 评论 -
【caffe】caffe框架中的train_val.prototxt文件和deploy.prototxt文件区别和转换
train_val.prototxt文件和deploy.prototxt文件区别和转换1.train_val.prototxt文件和deploy.prototxt文件介绍。train_val.prototxt:训练与测试使用的网络结构文件deploy.prototxt:模型构造文件。用于实际场景使用时的网络结构文件。这两个文件是caff...原创 2019-05-07 10:33:17 · 681 阅读 · 0 评论 -
【caffe】可视化网络的训练过程的loss和accuracy(效果优于官方绘制脚本)
前言:我们在训练网络的时候,网络训练完毕后,需要可视化训练过程中loss和accuracy的变化情况,方便了解整个训练过程模型的优化情况。(之前使用caffe自带的脚本绘制曲线图,出现各种报错,遂采用python脚本来绘制曲线。)1、保存训练过程的log。我们在执行训练指令时,可以通过以下方式保存训练日志。#这是sh指令,最后一句为设置保存训练日志/data_1/caffe-mas...原创 2019-10-28 14:37:49 · 700 阅读 · 0 评论 -
【caffe】使用python脚本训练分类模型,并可视化Loss和Accuracy.
PS:这里不是使用caffe默认的终端调用训练指令,而是使用python脚本来进行网络训练,并在网络训练完毕后显示训练过程的loss和accuracy。详细脚本如下:1)需要修改的变量:cafferoot=‘caffe根目录’solver='网络solver文件'loss和accuracy为自己网络文件中对应层的名称。2)其中绿色、黄色和红色曲线分别代表训练loss、测试l...原创 2019-10-28 10:34:22 · 560 阅读 · 0 评论 -
【caffe】caffe的solver中不同的学习策略(lr_policy)
在caffe源码的caffe-master/sec/caffe/proto/caffe.proto下记录了不同的学习策略的计算方法:// The learning rate decay policy. The currently implemented learning rate // policies are as follows: - fixed: always retu...转载 2019-08-08 14:48:25 · 170 阅读 · 0 评论 -
【caffe】SSD网络模型中solver.prototxt文件详解
train_net: "models/VGGNet/VOC0712/SSD_300x300/train.prototxt"test_net: "models/VGGNet/VOC0712/SSD_300x300/test.prototxt"test_iter: 619 #测试时的迭代次数test_interval: 10000 #每10000次迭代测试一次base_lr: 0.0...原创 2019-07-19 17:19:06 · 670 阅读 · 0 评论 -
【caffe】ubuntu系统下caffe的create_imagenet.sh(用于生成lmdb文件)与train_caffe.sh(训练指令)详解
1.create_imagenet.shcreate_imagenet.sh是ubuntu下caffe用来将图片数据转换成lmdb格式文件的指令。下面是create_imagenet.sh中的具体内容,其中本文此处设置的路径为windows下,在ubuntu下改成ubuntu对应的路径即可。#!/usr/bin/env sh# Create the imagenet lmdb in...原创 2019-05-08 11:34:13 · 614 阅读 · 0 评论 -
【caffe】Windows下caffe框架中生成均值文件
均值文件的意义图片减去均值后,归一化后,再进行训练和测试,能够提高速度和精度。因此,一般需要在训练时减去均值,均值文件的计算主要有两种方式:1)第一种方法:直接将均值设置为128,但若遇到一些填充过的样本,那么均值就会和128相差较多,这种情况下就要用第二种方法。2)第二种方法:实际上就是计算所有训练样本的平均值,计算出来后,保存为一个均值文件,在以后的测试中,就可以直接使用这个均值来...原创 2019-04-25 17:03:42 · 397 阅读 · 0 评论 -
【caffe】Windows下将文件夹中图像样本生成LMDB格式文件
1.创建convert_train_lmdb.bat文件,用来生成lmdb格式文件。.bat文件中具体内容:D:\\caffe-master\\Build\\x64\\Release\\convert_imageset.exe --gray --resize_width=144 --resize_height=144 D:/caffe-windows/Num_train/train/ D:...原创 2019-04-25 10:36:48 · 327 阅读 · 0 评论 -
【caffe】Ubuntu16.4系统安装caffe CPU版 (踩坑记录)
Ubuntu16.4系统安装caffe CPU版 (踩坑记录)PS:文章内容流程参考自以下文章,其中根据实际出现的问题,对其中的步骤做了增加。参考文章链接:https://blog.csdn.net/u010193446/article/details/53259294感谢作者:莫欺、少年穷一、检查自己电脑是否具有GPU 通常Caffe在计算时有两种模式可以选择,CPU...原创 2019-04-18 23:45:24 · 932 阅读 · 0 评论 -
【caffe】caffe中Solver文件简介
# The train/test net protocol buffer definitionnet: "examples/vggface/vggface_train_test.prototxt"# test_iter specifies how many forward passes the test should carry out.test_iter: 15 #设置迭代测试的次数...原创 2019-04-02 15:04:31 · 136 阅读 · 0 评论 -
【caffe】Windows10+Cuda9.0+Cudnn7.0下配置GPU版caffe
第一步:下载所需软件1、下载CUDA9.0,也是目前的最新版本。2、下载CUDNN V7.0。3、下载Caffe。 第二步:安装和配置参考网上的大多数教程,对软件进行安装配置。1、安装CUDA9.0,安装过程很简单。2、CUDA9.0安装完成后,打开CUDNN文件,将里面的bin、lib、include三个文件夹复制到CUDA9.1的安装路径中与CUDA9.1本...原创 2018-10-27 21:31:48 · 3629 阅读 · 13 评论 -
【caffe】Windows下使用GPU版caffe训练minist网络模型
Caffe的GPU版训练minist模型步骤1.下载训练数据。2.将训练数据生成lmdb文件1)建立下面两个脚本文件,用来将上面的4个数据转换为lmdb文件。其中脚本文件中的内容如下:文件:convert-mnist-data-test.bat..\..\..\Build\x64\Release\conver...原创 2018-10-27 21:08:43 · 648 阅读 · 0 评论 -
【TF】windows10 安装TensorFlow1.6教程
安装TensorFlow一共需要安装以下几个文件:1、Python3.52、cuda9.03、cudnn-9.0-windows10-x64-v7.zip(记住这里的cudnn版本要跟cuda相对应,不然会报错!)第一步:安装python3.5.去官网或其他地方下载好python3.5之后,直接一路点下一步就OK。可以在选安装路径的地方改成D盘或其他盘。这里给个安装详细教程链接https://bl...原创 2018-06-08 21:11:14 · 2591 阅读 · 0 评论