"如何用泡菜保存 keras StringLookup 对象？"

Question

我有一个计算机视觉模型来破解验证码。 在运行模型之前，我必须使用以下代码映射图像标签：

# Mapping characters to integers
  char_to_num = layers.StringLookup(
  vocabulary=list(characters), mask_token=None)


# Mapping integers back to original characters
  num_to_char = layers.StringLookup(
  vocabulary=char_to_num.get_vocabulary(), mask_token=None, invert=True)

另外， char_to_num对象的对象类型是： <keras.layers.preprocessing.string_lookup.StringLookup object at 0x00000269799D2C40>

我的问题是我想保存这个映射，以便以后加载模型时可以使用它。 有谁知道这是怎么做到的吗？ 也许用泡菜？

Answer 1

char_to_num.get_vocabulary()<\/code>是一个列表，您可以将其保存在文本文件中。 在推理期间读取文件并创建num_to_char<\/code> 。 您可以参考我在app.py<\/code> space OCR for CAPTCHA<\/a>上创建的演示中的 app.py

"