
tensorflow
晚归
勿以事小而不为
展开
-
BERT TextClassify Estimator预测速度慢的问题
在使用BERT做文本分类时,在使用tf.estimator.Estimator做预测时,速度总是很慢,单纯的一个生成器解析都很耗费时间;无论是使用ckpt和pb预测总是会变成对ckpt模型的预测,因此,本文重写了pb的预测代码:BERT模型训练完以后得到的文件有:...原创 2020-05-13 11:24:03 · 2161 阅读 · 3 评论 -
场景文本检测—CTPN算法
场景文本检测—CTPN算法何为OCR ?OCR的全称为“Optical Character Recognition” 中文翻译为光学字符识别。它是利用光学技术和计算机技术把印在或写在纸上的文字读取出来,并转换成一种计算机和人都能够理解的形式的过程。先来看一个简单的OCR的流程:第一步,通过图像信息采集(一般就是相机),得到包含了待识别字符的图像,分析结构。第二步,运用阈值...转载 2019-03-13 14:19:57 · 1056 阅读 · 0 评论 -
tensorflow指定使用哪块GPU运行程序
我认为最直观的一种方式:import os# 使用第三块GPUos.environ["CUDA_VISIBLE_DEVICES"] = "2"其他方式:第一种:在运行 py 文件时设置:CUDA_VISIBLE_DEVICES=0,1 python3 gpu_test.py第二种:动态分配显存:# 1.让TensorFlow按需分配显存config =...原创 2019-03-28 13:47:50 · 3465 阅读 · 0 评论