如何创建与FSNS数据集格式相同的数据集?

问题描述

我正在研究这个基于 TensorFlow 的 您可以使用以下代码段创建一个与 FSNS 兼容的 tf.Example proto:

char_ids_padded, char_ids_unpadded = encode_utf8_string(文本、字符集、长度、null_char_id)示例 = tf.train.Example(features=tf.train.Features(特征={'图像/格式':_bytes_feature("PNG"),'图像/编码':_bytes_feature(img.tostring()),'图像/类':_int64_feature(char_ids_padded)，'图像/未填充_类':_int64_feature(char_ids_unpadded)，'高度':_int64_feature(img.shape[0]),'宽度':_int64_feature(img.shape[1]),'orig_width': _int64_feature(img.shape[1]/num_of_views),'图像/文本':_bytes_feature(text)}))

I'm working on this project based on TensorFlow.

I just want to train an OCR model by attention_ocr based on my own datasets, but I don't know how to store my images and ground truth in the same format as FSNS datasets.

Is there anybody also work on this project or know how to solve this problem?

解决方案

The data format for storing training/test is defined in the FSNS paper https://arxiv.org/pdf/1702.03970.pdf (Table 4).

To store tfrecord files with tf.Example protos you can use tf.python_io.TFRecordWriter. There is a nice tutorial, an existing answer on the stackoverflow and a short gist.

Assume you have an numpy ndarray img which has num_of_views images stored side-by-side (see Fig. 3 in the paper):and a corresponding text in a variable text. You will need to define some function to convert a unicode string into a list of character ids padded to a fixed length and unpadded as well. For example:

char_ids_padded, char_ids_unpadded = encode_utf8_string(
   text='abc', 
   charset={'a':0, 'b':1, 'c':2},
   length=5,
   null_char_id=3)

the result should be:

char_ids_padded = [0,1,2,3,3]
char_ids_unpadded = [0,1,2]

If you use functions _int64_feature and _bytes_feature defined in the gist you can create a FSNS compatible tf.Example proto using a following snippet:

char_ids_padded, char_ids_unpadded = encode_utf8_string(
   text, charset, length, null_char_id)
example = tf.train.Example(features=tf.train.Features(
  feature={
    'image/format': _bytes_feature("PNG"),
    'image/encoded': _bytes_feature(img.tostring()),
    'image/class': _int64_feature(char_ids_padded),
    'image/unpadded_class': _int64_feature(char_ids_unpadded),
    'height': _int64_feature(img.shape[0]),
    'width': _int64_feature(img.shape[1]),
    'orig_width': _int64_feature(img.shape[1]/num_of_views),
    'image/text': _bytes_feature(text)
  }
))

这篇关于如何创建与FSNS数据集格式相同的数据集?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！