Error: cannot read as File when loading model

Error: cannot read as File when loading model

我想我犯了一个新手错误,但我很难弄清楚哪里出了问题。

错误:

C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing>node classify rlc.jpg (node:38620) UnhandledPromiseRejectionWarning: Error: cannot read as File: "model.json" at readFile (C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\filereader\FileReader.js:266:15) at FileReader.self.readAsText (C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\filereader\FileReader.js:295:7) at C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\@tensorflow\tfjs-core\dist\io\browser_files.js:226:36 at new Promise (<anonymous>) at BrowserFiles.<anonymous> (C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\@tensorflow\tfjs-core\dist\io\browser_files.js:159:39) at step (C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\@tensorflow\tfjs-core\dist\io\browser_files.js:48:23) at Object.next (C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\@tensorflow\tfjs-core\dist\io\browser_files.js:29:53) at C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\@tensorflow\tfjs-core\dist\io\browser_files.js:23:71 at new Promise (<anonymous>) at __awaiter (C:\Users\Awesome\Google Drive\Source\Programming\JS\Testing\node_modules\@tensorflow\tfjs-core\dist\io\browser_files.js:19:12) (node:38620) UnhandledPromiseRejectionWarning: Unhandled promise rejection. This error originated either by throwing inside of an async function without a catch block, or by rejecting a promise which was not handled with .catch(). To terminate the node process on unhandled promise rejection, use the CLI flag--unhandled-rejections=strict(see https://nodejs.org/api/cli.html#cli_unhandled_rejections_mode). (rejection id: 2) (node:38620) [DEP0018] DeprecationWarning: Unhandled promise rejections are deprecated. In the future, promise rejections that are not handled will terminate the Node.js process with a non-zero exit code.

代码:

const tf = require('@tensorflow/tfjs');
const tfnode = require('@tensorflow/tfjs-node');
const tmImage = require('@teachablemachine/image');
const fs = require('fs');
global.FileReader = require('filereader');
const uploadModel =  "model.json"
const uploadWeights = "weights.bin"
const uploadMetadata = "metadata.json"

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 model = await tmImage.loadFromFiles(uploadModel,uploadWeights,uploadMetadata);
    const predictions = await model.predict(image);
    console.log('Classification Results:', predictions);
}

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

imageClassification(process.argv[2]);

文件结构:

/Testing
  /node_modules
  classify.js
  metadata.json
  model.json
  package-lock.json
  rlc.jpg
  weights.bin

背景:尝试将我所学的部署图像分类模型构建在可教机器中,使用原生 javascript 并将其适应节点 js。我是一名新手,正在为我所遵循的所有教程所基于的节点和浏览器之间的环境差异绊倒。

我正在关注的教程:

库需要 loadFromFiles 函数 documentation in github 中的文件。 默认情况下,您不能在节点中使用文件 browser API。 所以你需要在节点环境中以某种方式填充它,检查这些库 node-fetch/fetch-blob node-file-api/file-api

file-api 的用法示例:

const fs = require('fs');
const path = require('path');
const FileAPI = require('file-api');

const uploadModel =  "model.json"
const uploadModelPath = path.join(process.cwd(), uploadModel);

// polyfill
Object.keys(FileApi).forEach(key => { 
    process[key] = FileApi[key];
})

const uploadModelFile = new File({ 
  buffer: fs.readFileSync(uploadModelPath)
});

虽然这个库很旧,但它可能无法正常工作,您可以尝试搜索其他 polyfill 库或自己编写。 您可以在 tensor flow source code 中查看文件的读取方式,或者您可以分叉并添加使用节点文件的可能性。