"如何用泡菜保存 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

"