yoyoshuang2013-CSDN博客

原创 Pytorch 模型转tensorRT 记录

搜索到的pytorch 模型转tensorrt 模型有2种方案：直接将pytorch 模型转成trt , 参考：https://github.com/NVIDIA-AI-IOT/torch2trt来自 https://blog.csdn.net/debug_moner/article/details/90313514，但是这个方法实际操作的时候没有成功，遇到了一个问题，也是github上很多人都遇到的 https://github.com/NVIDIA-AI-IOT/torch2trt/issu...

2020-12-31 10:51:33 635 2

原创无root账号下用conda 安装软件到指定磁盘目录

创建conda环境, 有两种方式，下面这种可以指定创建路径conda create --prefix=/disk1/myname/conda-env3 python=3.6.6另外一种默认安装到conda的安装路径下：conda create -n py36 python=3.6这里使用了第一种。激活的时候命令为：source activate /disk1/myna...

2019-03-27 09:16:33 2444

原创 docker踩坑记录二

昨天安装了docker后，需要启动tf镜像命令docker run --runtime=nvidia -it tensorflow/tensorflow:1.5.0-gpu-py3 bash然后就一直报错：requirement error: unsatisfied condition: cuda >= 9.0\\\\n\\\"\"": unknown.明明已经装好了cu...

2019-02-15 11:56:45 1310 1

原创 nvidia-docker 安装报错记录

最近因为业务需要，很多服务要用docker部署，于是开始研究docker的使用。代码是python，使用的深度学习框架为tensorflow, 按照官网说明，需要先安装 Docker 和 nvidia-docker。其中Docker的安装比较简单，基本就是参照了这篇文档：https://yeasy.gitbooks.io/docker_practice/install/centos.html...

2019-02-14 16:19:07 12459

原创 ocr 训练记录 3

继续。第17次训练效果一般，所以就不记录了，不过这证实了黑白化对提升准确率没有太大帮助，于是放弃。【第18次训练】在第13次训练的基础上，新增了干扰线，按照40%的出现概率绘制干扰线，syn 与 gan 都是灰度图进行训练（放弃了黑白图）迭代次数：51200在真实数据上测试结果：总计图片数：513字符准确率：97.2%词准确率：85.1%全对的图片数：437错1-2个图片数：67错大于2个的图片数...

2018-06-11 13:35:15 263 2

原创新增真实数据测试记录

周一去财务那边拍摄了400多张图，都是真实数据。准确来说是100多张票用不同型号的手机分别拍摄一次得到的。其中 iPhone 6 s 拍摄145张，xiaomi_mix_2s 拍摄138张，xiaomi_note_lite拍摄 131张总计 145+138+131 = 414张【总体识别效果】全票面100% 正确数：129 （31%）total mean acc:87.95%(383, 327, ...

2018-06-08 11:06:02 352

原创 Linux 常用命令

查看文件创建/修改时间：stat ***.txt

2018-06-05 13:58:17 118

原创记一次代码重构

最近训练ocr，从直方图规范，到像素规范，到二值化调整，试了很多种办法，测试代码越写越复杂，需要对不同测试情况配置参数，怕后面还有新的增加，于是决定重构一下函数，结果重构后，无法复现以前的实验结果了。桑心～怕越改越乱，今天静下心来好好看了一下，哈哈，找到问题所在了，在此记录一下。重构后的代码：def img_to_fit(img,source,flags): hist_norm_flag,i...

2018-05-30 16:29:46 187

原创 OCR 训练记录2

上一篇太长了，分开写【第13次训练】增加拉伸压缩的多样性：scale_list1 = [0.7,0.75,0.8,0.85,0.9,0.95]scale_list2 = [1/x for x in scale_list1 ]结束标志：平均acc达到0.999在真实数据上测试结果：字符准确率：95%word 准确率：75%真实样本总数：124全部正确数：93错1-2个数：30其他：1【第13.1次训...

2018-05-24 10:24:51 621 4

原创 OCR 训练记录

【第一次训练】训练数据：syn训练数量：51200在真实数据上测试结果：字符准确率：67%word 准确率：5%【第二次训练】训练数据：syn + gan训练数量：syn 51200， gan 51200是否提前结束：是结束标志：11个字符平均acc 达到98%在真实数据上测试结果：字符准确率：75.3%word 准确率：14.5%真实样本总数：124全部正确数：18错1-2个数：27其他：79...

2018-05-17 13:23:49 1219

原创 keras 数据生成器使用小结

今天尝试用keras实现SimGan，在读入数据时遇到了几个坑，记录一下。# coding: utf-8import osimport sysimport kerasfrom keras import applicationsfrom keras import layersfrom keras import modelsfrom keras import optimizersfro...

2018-04-26 20:44:22 2510

原创将已有 image 文件转换成 tfrecord 文件

最近在做OCR的一个项目，手上的训练图片不够用，于是想通过 CGAN 的方法生产一些训练图片，遇到的第一个问题是将已有图片保存成 tfrecord 文件。代码如下：# coding: utf-8from __future__ import absolute_importfrom __future__ import divisionfrom __future__ import print_fu...

2018-04-19 10:45:01 1008

yoyoshuang2013的博客