megengine.functional.nn.correlation

correlation(data1, data2, kernel_size=1, max_displacement=1, stride1=1, stride2=1, pad_size=0, is_multiply=True)[源代码]

计算两特征图之间的相关系数。

参数

  • data1 (Tensor) – 格式为 NCHW 的特征图1

  • data2 (Tensor) – 格式为 NCHW 的特征图2

  • kernel_size (int) – 非负奇整数，表示 patch 的大小，即计算时邻域大小。默认值为 1

  • max_displacement (int) – 非负整数，计算时的窗口大小。默认值为 1

  • stride1 (int) – 非负整数，特征图 1 上取值的步长。其值决定输出的宽和高。默认值为 1。

  • stride2 (int) – 非负整数，在特征图 2 中取特征图 1 的邻域的步长。默认值为1。

  • pad_size (int) – 非负整数，在四个方向分别对两输入补 0 的数量

  • is_multiply (bool) – 布尔值，设置为 True 时累积对应点的乘积，否则累积对应点差值的绝对值。默认为 True。

返回类型

Tensor

上一页

megengine.functional.nn.nms

下一页

megengine.functional.nn.nvof