tfjs icon indicating copy to clipboard operation
tfjs copied to clipboard

[WebGL backend] Wrong results for AutoML Object model

Open Linchenn opened this issue 3 years ago • 0 comments

When executing AutoML Object model with the following code snippet, it works for 3.19.0 version of WebGL backend, but it does not work for the current github repo.

const url =
          'https://storage.googleapis.com/tfjs-testing/tfjs-automl/object_detection/model.json';
const model = await tf.loadGraphModel(url);
const OUTPUT_NODE_NAMES =
          ['Postprocessor/convert_scores', 'Postprocessor/Decode/transpose_1'];
const feedDict = {ToFloat: tf.zeros([1, 224, 224, 3])};

await model.executeAsync(feedDict, OUTPUT_NODE_NAMES);

Linchenn avatar Aug 08 '22 19:08 Linchenn