yiweibian的博客

学以致用,帮人如帮己

tensorflow执行到session.run的时候发现内存泄漏原因排查

场景:加载训练好的模型进行推理 代码大体实现流程:输入—》模型—》结果 模型为一个单独的类,其中在调用模型的时候需要import 类,然后加载模型在类中init ()函数。也就是说当运行代码的时候先加载模型,然后模型等待别的函数调用进行推理。 实际上是每次都需要调用session.run()函数。...

2018-11-19 17:02:56

阅读数 159

评论数 0

window 远程使用jupyter notebook

远程主机:linux 本地主机:windows 10 安装jupyter pip install jupyter notebook 配置jupyer $ python #输入以下内容 from notebook.auth import passwd; passwd() #需要设置两次密码...

2018-10-31 09:55:02

阅读数 40

评论数 0

Tvm(1) SSD 模型

TVM 就不多介绍了,本文涉及TVM的安装和SSD 模型的测试。 TVM 安装 $ git clone --recursive https://github.com/dmlc/tvm $ sudo apt-get update $ sudo apt-get install -y pyt...

2018-06-28 11:08:25

阅读数 1096

评论数 1

ocr(2)-crnn-mxnet-chinese-text-recognition

ubuntu 16.04 GTX1080 Mxnet 最新 pytho 2.7 1. 安装Mxnet Mxnet 按照参考官网,选择源码GPU安装。 2. 安装warp-ctc 安装warp-ctc $ git clone https://github.com/bai...

2018-06-05 11:23:48

阅读数 482

评论数 0

ocr(1)-Attention-ocr的安装测试

Attention-ocr是一种端到端的图像文字识别方法。其输入是含有文字的图片,输出是文字。整个出来过程是: 图像—-CNN特征提取—Encoder—Visual Attention—Decoder 以下是基于Github上的Attention-OCR进行安装测试 Ubuntu...

2018-06-04 16:59:54

阅读数 2138

评论数 0

py-faster-rcnn训练过程中遇到的一些错误

以下问题都是根据参考文献修改 -1.AttributeError: ‘module’ object has no attribute ‘text_format’ 解决方法:在/home/xxx/py-faster-rcnn/lib/fast_rcnn/train.py的头文件导入部分加上 :i...

2018-05-22 10:48:02

阅读数 142

评论数 0

遗传算法(1)--初步认识

最近学习了遗传算法,尝试着把它按照理解写下来。对于遗传算法从名字上就可以看出来是和生物有关的东西,可以认为是和生物进化有关系,在生物进化里面又有一句话叫:物竞天择,适者生存。也就是说受环境的影响,最终会留下那些能适应环境的生物。如果从生物学里面进行考虑可能会涉及到 一堆的概念名词,如:染色体,个体...

2018-05-09 16:16:38

阅读数 160

评论数 0

FPN(Feature Pyramid Networks for Object Detection)安装与训练

FPN主要使用特征金字塔网络来融合多层特征,改进了CNN特征提取。其在小目标上取得了很大的进步。更多细节可以阅读论文。 下面分两个内容介绍:1. 安装 2. 训练测试1.安装对于FPN有好几个框架实现的版本,如:caffe,mxnet 等。本文是对caffe 版本的FPN安装。 1.1 ...

2017-11-08 14:24:05

阅读数 4089

评论数 13

SSD 训练自己的数据

忽略细节较多,只做一个简单的说明。 系统:ubuntu16.04 python2.7 opencv3.3.3 主要内容1.安装 ssd 2.准备数据 3.训练1.安装 ssd可以根据官网的Toturial安装#1.get the code git clone https://github...

2017-10-16 12:54:31

阅读数 721

评论数 0

Opencv对视频进行目标检测

在Opencv3.3版本中集成了deeplearning功能。其实现了对caffe和tensorflow两个框架的推理,但不支持训练。本文使用caffe训练的文件对目标进行检测。整个思路是首先读取视频文件,然后加载模型文件,最后读取到视频的每一帧对其进行检测。 系统: ubuntu16.04 py...

2017-10-12 13:42:42

阅读数 3013

评论数 0

git(1)上传自己的项目到github上面

假如有一个存在的项目test-project,此时你有一个github帐号:test@github.com,想把test-project上传到自己帐号里面,那么只需要以下几步: 建立系统和github的ssh通信 在github创建项目 上传到github 1. 建立系统和github的ssh通信...

2017-08-17 18:00:10

阅读数 218

评论数 0

Tensorflow模型转caffe

整个过程可以分两步: 找到对应的层名字(如:tensorflow 模型中的conv1_1和caffe中的conv1_1)。 把weights 和biases 中的值赋给相对应的层。赋值的时候要注意Tensorflow和caffe之间通道不一样,这里还是做转换。 实现 实现的代码地址

2017-08-15 10:58:31

阅读数 4089

评论数 0

Mxnet图片分类(4)利用训练好的模型进行测试

利用训练好的模型测试只需要把模型和数据准备好。系统: ubuntu14.04 Mxnet: 0.9041.模型和数据准备2.模型加载测试import mxnet as mx sym,arg_params,aux_params = mx.model.load_checkpoint('vggnew'...

2017-05-24 16:57:23

阅读数 3807

评论数 0

Mxnet图片分类(3)fine-tune

使用fine-tun的方式训练模型的话首先需要下载相应的模型,然后按照自己的数据集修改相应的类别,最后训练。 系统: ubuntu14.04 Mxnet: 0.9041.数据准备train_iter = "/mxnet/tools/train-cat.rec"val_iter...

2017-05-24 16:35:43

阅读数 2822

评论数 0

Mxnet图片分类(2)训练模型

训练模型可以利用自定义的模型进行也可以采用fine-tune的方法。这里先介绍如何自定义模型进行训练。 系统: ubuntu14.04 Mxnet: 0.9041.数据准备mxnet数据集的生成可以参考上一篇文章这里从代码来分析:#数据路径 train_iter = "/mxnet/...

2017-05-24 15:20:46

阅读数 1999

评论数 0

Mxnet图片分类(1)准备数据集

Mxnet做图片分类训练的时候提供了多种输入,这里介绍.rec数据的制作。mxnet版本:9.041.收集图片这里收集了两类图片,猫和狗的图片。 第一类 第二类所有的图片放到两个文件夹中 2.生成mxnet训练需要的数据格式如果mxnet自带的im2rec.py有问题,这里用的是另外的文...

2017-05-24 11:58:45

阅读数 2693

评论数 0

Mxnet(4)-fcn训练自己的数据

fcn为全卷积网络,对于算法的介绍可以找到很多资料,本篇在mxnet框架下用自己的数据,或者说制作自己的数据来进行模型的训练和测试。假设已经安装好有关的Mxnet 的环境,其中在Mxnet安装包下的`example/fcn-xs`下找到相关的文件。这里也假设你已经跑过fcn-xs下的`image_...

2017-04-18 17:32:21

阅读数 4188

评论数 4

Mxnet(3)-SSD训练自己的数据

整个ssd的安装和训练可以按照官方的例子。需要注意的是需要编译ssd相关的代码,可以通过两种方式进行: 直接把ssd目录下的operator中的文件复制到src/operator,然后再编译 设置config.mk,把EXTRA_OPERATORS改成EXTRA_OPERATORS = exam...

2017-04-17 10:54:50

阅读数 6661

评论数 4

Mxnet(2)---faster-rcnn制作自己的数据集进行训练

Mxnet自带有faster-rcnn的例子,但是如果要用自己的数据进行训练可能需要作一些更改,一个是类别的数目,一个数据的标签。其实它的修改方式和py-faster-rcnn差不多。 **1.环境的安装可以参考**:Mxnet—faster-rcnn环境安装**2.制作数据集** ...

2017-03-28 12:51:33

阅读数 5480

评论数 1

Mxnet---faster-rcnn环境安装

官方版本已经有faster-rcnn的实现,所以这里从官方下载mxnet并安装编译。安装环境: Ubuntu14.04 GPU gtx1080 1.mxnet的安装 这里还是从两种情况分析,一种是之前已经安装好了mxnet,但是没有安装编译faster-rcnn,第二种是没有安装过任何...

2017-03-14 14:23:52

阅读数 3386

评论数 0

提示
确定要删除当前文章?
取消 删除
关闭
关闭