Tensorflow.js tf.image.nonMaxSuppressionPadded()用法及代码示例

Tensorflow.js是Google开发的开放源代码库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.image.nonMaxSuppressionPadded() 函数用于基于 iou 异步执行限制框的非最大抑制，即联合的交集以及填充结果的机会。

用法:

tf.image.nonMaxSuppressionPadded(boxes, scores, maxOutputSize, 
    iouThreshold?, scoreThreshold?, padToMaxOutputSize?)

Parameters: 

• boxes:所述的二维张量，其配置为 [numBoxes, 4]。并且每次访问都是 [y1, x1, y2, x2]，允许 (y1, x1) 和 (y2, x2) 是限制框的边。它可以是 tf.Tensor2D、TypedArray 或 Array 类型。
• scores:规定的一维张量，前提是框得分是配置 [numBoxes]。它是 tf.Tensor2D、TypedArray 或 Array 类型。
• maxOutputSize:它是要拣选的指定箱子的最大数量。它是类型号。
• iouThreshold:规定的浮点数表示阈值，以决定规定的框是否与 IOU 相交太多。它应该在 [0, 1] 的中间。默认值为 0.5，即 50% 的框相交。它是可选的，类型为 number。
• scoreThreshold:这是规定的阈值，以便根据规定的分数决定在哪些时间删除框。默认值为 -inf，即允许每个分数。它是可选的，类型为 number。
• padToMaxOutputSize:它是布尔类型的可选参数。默认值为 false。如果为真，则结果 selectedIndices 的维度将填充到 maxOutputSize。

返回值：它返回 {[name:string]:tf.Tensor}。

范例1：在此示例中，我们将使用 2d 张量、分数和 maxOutputSize 参数。

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling image.nonMaxSuppressionPadded() method
const output = tf.image.nonMaxSuppressionPadded(
    tf.tensor2d([1, 2, 3, 4, 2, 4, 6, 7], 
    [2, 4]), [1, 1], 4
);
  
// Printing output
console.log(output);

输出：

{
  "selectedIndices":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[
      2
    ],
    "dtype":"int32",
    "size":2,
    "strides":[],
    "dataId":{
      "id":22
    },
    "id":22,
    "rankType":"1",
    "scopeId":12
  },
  "validOutputs":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[],
    "dtype":"int32",
    "size":1,
    "strides":[],
    "dataId":{
      "id":23
    },
    "id":23,
    "rankType":"0",
    "scopeId":12
  }
}

范例2：在此示例中，我们将使用浮点数数组 iouThreshold、scoreThreshold 以及 padToMaxOutputSize。

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining an array of floats
const arr = [[11.1, 2.3, 7.3, 6.4], [3, 6]]
  
// Calling image.nonMaxSuppressionPadded() method
const res = tf.image.nonMaxSuppressionPadded(
    arr, [2.1, 0], 100, 0.5, 1, true);
  
// Printing output
console.log(res);

输出：

{
  "selectedIndices":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[
      2
    ],
    "dtype":"int32",
    "size":2,
    "strides":[],
    "dataId":{
      "id":42
    },
    "id":42,
    "rankType":"1",
    "scopeId":22
  },
  "validOutputs":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[],
    "dtype":"int32",
    "size":1,
    "strides":[],
    "dataId":{
      "id":43
    },
    "id":43,
    "rankType":"0",
    "scopeId":22
  }
}

参考： https://js.tensorflow.org/api/latest/#image.nonMaxSuppressionPadded