Невозможно загрузить обученную Python модель в tenorflow.js

1

Я получаю следующие ошибки, когда я пытаюсь загрузить модель, которую я тренировал в Python, когда я использую функцию loadModel loadModel() tensorflow.js:

Не удалось загрузить ресурс: net :: ERR_NAME_NOT_RESOLVED

Неподготовлено (в обещании) TypeError: Не удалось получить

Ниже приведен файл pred.js

console.log ("hello");

let model;
(async function () {
    model = await tf.loadModel("http://keras_model/model.json");
    $(".progress-bar").hide();
    console.log("it works");
})();

Структура каталогов:

main
  -dataset  (contains images for training the model) 
  -training_scripts (python scripts to train the model)
  -user_interface
     --server.js (server made using node.js(and express))
     --static (this folder contains the trained keras model)
         --index.html (html file to be served)
         --predict.js 
         --keras_model(this folder contains the model.json file)

Любая помощь будет оценена!

  • 0
    Ваш URL должен быть абсолютным, если вы используете http:// качестве типа файла
  • 0
    Или, если вы используете узел и хотите загрузить локальный файл, вы должны использовать файл типа file:///
Показать ещё 4 комментария
Теги:
tensorflow
tensorflow.js

2 ответа

1
Лучший ответ

Если вы хотите загрузить локальные файлы в tfjs, вам нужно использовать файл типа file:/// и для этого вам понадобится расширение узла tfjs. Вы можете загрузить его, установив и потребовав node-fetch в вашу программу.

Поделиться
0

Вы также можете использовать обработчик fileSystem представленный в tfjs-node:

const tf = require("@tensorflow/tfjs");
const tfn = require("@tensorflow/tfjs-node");
const handler = tfn.io.fileSystem("./path/to/your/model.json");
const model = await tf.loadModel(handler);
Поделиться

Ещё вопросы

Сообщество Overcoder
Наверх
Меню