このコードを使用して画像分類を試みています。ユーザーが 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