TensorFlow Тензор к тензору TensorFlow.js - PullRequest
0 голосов
/ 05 июля 2019

Я пытался взять тензор из вывода ConvNet и использовать его в другой модели в TF.js. Есть ли способ сохранить тензор TF в файл и загрузить этот файл в TF.js и получить тензор обратно?

1 Ответ

0 голосов
/ 05 июля 2019

Тензор может быть сохранен в файл. Однако для его загрузки требуется сервер, поскольку браузер не имеет прямого доступа к файловой системе, за исключением случаев использования input type file.

Чтобы сохранить тензор в файл, значения тензора должны быть сначала загружены с использованием data или dataSync.

const tensor = tf.tensor([1, 2])
// download values
const values = tensor.dataSync()

Чтобы увидеть, как вы можете сохранить values в файл, вы можете использовать следующий ответ

Чтобы загрузить сохраненный тензор, содержимое файла должно обслуживаться сервером с запросом http. Получив ответ на запрос,

const tensorRetrieved = tf.tensor(contentHttpResponse) // if needed parse the string response using JSON.parse

Другой вариант - использовать тензор потока.js на узле. Запрос http и ответ не понадобятся, поскольку узел может получить доступ к файловой системе.

...