
Tensorflow
abner15
这个作者很懒,什么都没留下…
展开
-
Tensorflow模型转caffe
整个过程可以分两步:找到对应的层名字(如:tensorflow 模型中的conv1_1和caffe中的conv1_1)。把weights 和biases 中的值赋给相对应的层。赋值的时候要注意Tensorflow和caffe之间通道不一样,这里还是做转换。 实现 实现的代码地址原创 2017-08-15 10:58:31 · 6705 阅读 · 5 评论 -
ocr(1)-Attention-ocr的安装测试
Attention-ocr是一种端到端的图像文字识别方法。其输入是含有文字的图片,输出是文字。整个出来过程是: 图像—-CNN特征提取—Encoder—Visual Attention—Decoder以下是基于Github上的Attention-OCR进行安装测试Ubuntu 16.04GPU:GTX1080Tensorflow 1.8.0python 2.7.12...原创 2018-06-04 16:59:54 · 3905 阅读 · 0 评论 -
tensorflow执行到session.run的时候发现内存泄漏原因排查
场景:加载训练好的模型进行推理代码大体实现流程:输入—》模型—》结果模型为一个单独的类,其中在调用模型的时候需要import 类,然后加载模型在类中init ()函数。也就是说当运行代码的时候先加载模型,然后模型等待别的函数调用进行推理。实际上是每次都需要调用session.run()函数。在这个过程中发现在循环调用session.run()出现了内存泄漏的问题。首先session 是se...原创 2018-11-19 17:02:56 · 3275 阅读 · 2 评论