Как восстановить актив файла Vocab, сохраненный с помощью Tensorflow SavedModel - PullRequest
Новая
       21

Как восстановить актив файла Vocab, сохраненный с помощью Tensorflow SavedModel

0 голосов
/ 28 августа 2018

Я работаю в блоге Tensorflow, чтобы создать CNN.

Это отличная статья, но она использует следующий код для сохранения и восстановления файла vocab в файловой системе: 

vocab_processor.save(vocab)
vocab_processor = learn.preprocessing.VocabularyProcessor.restore(vocab))

Я расширяю пример для использования подхода SaveModel для создания сохраненного_моделя.pd, и это работает нормально. В качестве следующего шага я хотел бы сохранить и восстановить файл vocab как актив.

Действие сохранения работает хорошо, и я вижу файл voab в моем каталоге / assets, такого же размера и содержания, как при использовании оригинального подхода.

У меня вопрос, как мне восстановить содержимое файла из каталога активов для передачи в метод VocabularyProcessor.restore?

Я нашел исчерпывающий пример класса загрузчика для обработки каждого актива и добавления его в словарь тензора актива, но это дает мне путь и имя файла, а не данные. 

for asset_any_proto in assets_any_proto:
    asset_proto = meta_graph_pb2.AssetFileDef()
    asset_any_proto.Unpack(asset_proto)
    tensor_name = asset_proto.tensor_info.name
if import_scope:
    tensor_name = "%s/%s" % (import_scope, tensor_name)
asset_tensor_dict[tensor_name] = os.path.join(
    compat.as_bytes(assets_directory),
    compat.as_bytes(asset_proto.filename))

Как мне добиться следующего: 

vocab_processor = learn.preprocessing.VocabularyProcessor.restore(RESTORED_ASSET))

возможно ли это с Tensorflow?

...