在自定义数据集上训练 tensorflow attention-ocr 的管道是啥?

     2023-04-17     292

关键词:

【中文标题】在自定义数据集上训练 tensorflow attention-ocr 的管道是啥?【英文标题】:what's the pipeline to train tensorflow attention-ocr on customized dataset?在自定义数据集上训练 tensorflow attention-ocr 的管道是什么? 【发布时间】:2018-10-01 08:56:44 【问题描述】:

我在***上阅读了一些关于attention-ocr的问题,其中大部分是关于特定步骤的实现细节。我想知道的是我们在自己的数据集上微调这个模型的管道。

据我所知,步骤应该是:

0) 我们应该先下载 FSNS 数据集吗?我试图绕过这一步并尝试仅在一张图像上运行推理,但它总是给我错误:“ImportError:没有名为'fsns”的模块。所以我想知道一旦我设置了自己的数据集,这个错误是否会消失。

1) 以与 FSNS 相同的格式存储我们的数据。 (本主题链接:How to create dataset in the same format as the FSNS dataset?、how to create cutomized dataset for google tensorflow attention ocr?)

2) 下载预训练的检查点(http://download.tensorflow.org/models/attention_ocr_2017_08_09.tar.gz)

3) 以某种方式修改“model.py”以适合您自己的目的。

4) 以某种方式修改“train.py”以使用 tensorflow 服务训练您自己的模块。

我现在仍处于这个项目的早期阶段(创建自己的数据集),对如何做以及下一阶段是什么感到困惑。

【问题讨论】:

【参考方案1】:

错误是由不正确的 Python 版本引起的。它们应该与 Python 2 一起运行,您只需更改“导入”语句即可解决此错误。尝试将“import fsns”更改为“from datasets import fsns”。

【讨论】:

在自己的数据集上训练 TensorFlow 对象检测

】在自己的数据集上训练TensorFlow对象检测【英文标题】:TrainTensorflowObjectDetectiononowndataset【发布时间】:2017-12-1121:27:26【问题描述】:在花了几天时间尝试完成这项任务后,我想分享一下我如何回答这个问题的经验:如何使用TS... 查看详情

在自定义数据集上使用 roboflow 对象检测 Yolov4 pytorch 模型时出现值错误

】在自定义数据集上使用roboflow对象检测Yolov4pytorch模型时出现值错误【英文标题】:ValueerrorwhileusingroboflowobjectdetectionYolov4pytorchmodeloncustomdataset【发布时间】:2021-12-1319:43:48【问题描述】:我们使用Roboflow进行对象检测,使用Yolov4... 查看详情

Tensorflow:在具有不同类别数量的新数据集上微调预训练模型

】Tensorflow:在具有不同类别数量的新数据集上微调预训练模型【英文标题】:Tensorflow:Finetunepretrainedmodelonnewdatasetwithdifferentnumberofclasses【发布时间】:2017-06-0415:24:17【问题描述】:如何在新数据集上微调TensorFlow中的预训练模型... 查看详情

机器学习笔记-使用tensorflowlite从头创建模型(代码片段)

一、概述        目标是使用TFLiteModelMakerLibrary创建TensorFlowLite模型。将在自定义数据集上微调预训练的图像分类模型,并进一步探索该库当前支持的不同类型的模型优化技术,并将其导出到TFLite模型。最后对创建的TFL... 查看详情

为啥在同一数据集上使用 tensorflow 和 keras 重新训练的 Inception V3 显示出不同的准确性?

】为啥在同一数据集上使用tensorflow和keras重新训练的InceptionV3显示出不同的准确性?【英文标题】:WhyInceptionV3retrainedusingtensorflowandkerasonsamedatasetshowsdifferentaccuracy?为什么在同一数据集上使用tensorflow和keras重新训练的InceptionV3显示... 查看详情

在自定义数据集上微调 MobileNet 时出现形状错误

】在自定义数据集上微调MobileNet时出现形状错误【英文标题】:ShapeErrorwhileFineTuningMobileNetOnACustomDataSet【发布时间】:2022-01-1611:10:50【问题描述】:我跟随deeplizard对MobileNet进行微调。我试图做的是从模型的第5层到最后一层抓取... 查看详情

获取 InvalidArgumentError:NewRandomAccessFile 在自定义图像数据集上使用 tf.data 时无法创建/打开

】获取InvalidArgumentError:NewRandomAccessFile在自定义图像数据集上使用tf.data时无法创建/打开【英文标题】:GettingInvalidArgumentError:NewRandomAccessFilefailedtoCreate/Openwhileworkingwithtf.dataonthecustomimagedataset【发布时间】:2019-04-0403:46:56【问题描... 查看详情

如何在 tensorflow 对象检测 API 中使用“忽略”类?

】如何在tensorflow对象检测API中使用“忽略”类?【英文标题】:Howtouse"ignore"classwithtensorflowobjectdetectionAPI?【发布时间】:2020-05-0307:11:08【问题描述】:我已经在自定义数据集上使用SSD(mobilenet-v1)训练了tensorflow对象检测模... 查看详情

AttributeError:“Tensor”对象在自定义损失函数中没有属性“numpy”(Tensorflow 2.1.0)

...r:“Tensor”对象在自定义损失函数中没有属性“numpy”(Tensorflow2.1.0)【英文标题】:AttributeError:\'Tensor\'objecthasnoattribute\'numpy\'incustomlossfunction(Tensorflow2.1.0)【发布时间】:2020-07-1817:06:16【问题描述】:我想用自定义损失函数训... 查看详情

TensorFlow:神经网络在训练集和测试集上的准确率始终为 100%

】TensorFlow:神经网络在训练集和测试集上的准确率始终为100%【英文标题】:TensorFlow:NeuralNetworkaccuracyalways100%ontrainandtestsets【发布时间】:2018-02-2114:09:53【问题描述】:我创建了一个TensorFlow神经网络,它有2个隐藏层,每个隐藏... 查看详情

将冻结图转换为 tensorflow-js 格式

】将冻结图转换为tensorflow-js格式【英文标题】:Convertfrozengraphtotensorflow-jsformat【发布时间】:2018-09-2217:22:37【问题描述】:我有一个使用Google对象检测API的SSD模型(在自定义数据集上训练)。我冻结了一个生成几个文件(包括... 查看详情

TensorFlow 检测 API 中的 SSD 锚点

】TensorFlow检测API中的SSD锚点【英文标题】:SSDanchorsinTensorflowdetectionAPI【发布时间】:2019-01-1511:45:13【问题描述】:我想在N×N图像的自定义数据集上训练SSD检测器。于是我挖了TensorflowobjectdetectionAPI,在COCO上找到了一个基于MobileN... 查看详情

Yolo 没有开始训练

...【发布时间】:2020-09-2910:31:37【问题描述】:我正在尝试在自定义数据集上训练Yolo,一切似乎都正常工作,但它只是没有训练。我按照https://github.com/AlexeyAB/darknet上的教程学习了两次,但得到了相同的结果./darknetdetectortraindata/obj... 查看详情

物体检测快速入门系列|01-基于tensorflow2.xobjectdetectionapi构建自定义物体检测器(代码片段)

这是机器未来的第1篇文章,由机器未来原创1.概述tensorflowobjectdetectionapi一个框架,它可以很容易地构建、训练和部署对象检测模型,并且是一个提供了众多基于COCO数据集、Kitti数据集、OpenImages数据集、AVAv2.1数据集和iNaturalist物... 查看详情

在 MNIST 集上使用 TensorFlow 进行预测的困境

】在MNIST集上使用TensorFlow进行预测的困境【英文标题】:DillemawithpredictionwithTensorFlowonMNISTset【发布时间】:2017-10-1014:03:54【问题描述】:目前我是TensorFlow的新手,我已经使用MNIST集训练了一个模型,现在我制作了一些带有数字的... 查看详情

TensorFlow Object Detection API 只训练一个类

】TensorFlowObjectDetectionAPI只训练一个类【英文标题】:TensorFlowObjectDetectionAPItrainsonlyoneclassfromseveral【发布时间】:2018-01-1413:20:28【问题描述】:我正在使用TF对象检测API在我自己的数据集上运行几个类的训练过程,但是经过相当多... 查看详情

训练后如何保存 MASK RCNN 模型?

...2019-11-1709:54:23【问题描述】:我正在使用matterport存储库在自定义数据集上训练MASKRCNN。我在训练中取得了成功。现在我想保存经过训练的模型并在Web应用程序中使用它来检测对象。训练后如何保存maskrcnn模型?请指导我。仓库的... 查看详情

tensorflow训练mnist数据集——卷积神经网络(代码片段)

   前面两篇随笔实现的单层神经网络 和多层神经网络, 在MNIST测试集上的正确率分别约为90%和96%。在换用多层神经网络后,正确率已有很大的提升。这次将采用卷积神经网络继续进行测试。 1、模型基本结构 ... 查看详情