4

このコードを使用して画像分類を試みています。ユーザーが image を指定すると、mobilenet モデルがロードされ、事前にトレーニングされた画像が表示されます。しかし、nodejs に mobilenet モデルをロードできません。私は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]);

node classify 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
4

0 に答える 0