Загрузите модель Mobil enet с Tensorflow. js in nodejs - PullRequest
0 голосов
/ 21 апреля 2020

Я пытаюсь классифицировать изображения с помощью этого кода. когда пользователь дает изображение, то для предварительного обучения будет загружена модель mobil enet. Но я не могу загрузить модель mobil enet в nodejs. Я использую последнюю версию nodejs. Итак, вот мой пробный код:

const tf = require('@tensorflow/tfjs');
const mobilenet = require('@tensorflow-models/mobilenet');
const tfnode = require('@tensorflow/tfjs-node');

const fs = require('fs');

const readImage = path => {

  const imageBuffer = fs.readFileSync(path);

  const tfimage = tfnode.node.decodeImage(imageBuffer);
  return tfimage;
}

const imageClassification = async path => {
  const image = readImage(path);
  const mobilenetModel = await mobilenet.load();
  const predictions = await mobilenetModel.classify(image);
}

if (process.argv.length !== 3) throw new Error('Incorrect arguments: node classify.js <IMAGE_FILE>');

imageClassification(process.argv[2]);

когда я запускаю с использованием узел классификации banana.jpg , это выдает мне эту ошибку. Как это будет обрабатываться?

(node:17038) UnhandledPromiseRejectionWarning: FetchError: request to https://storage.googleapis.com/tfhub-tfjs-modules/google/imagenet/mobilenet_v1_100_224/classification/1/model.json failed, reason: Client network socket disconnected before secure TLS connection was established
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...