昇腾社区首页
EN
注册

MDC_Correlation

输入

  • img0:
    • 是否必填:必填
    • 数据类型:float16
    • 参数解释:第一个输入张量,shape[1, 96, 48, 64]
    • 规格限制:NCHW
  • img1:
    • 是否必填:必填
    • 数据类型:float16
    • 参数解释:第二个输入张量,shape[1, 96, 48, 64]
    • 规格限制:NCHW

属性

  • pad:
    • 是否必填:非必填
    • 数据类型:uint32
    • 参数解释:意义同卷积,同时对两个featuremap实施
    • 规格限制:默认值20
  • kernel_size:
    • 是否必填:非必填
    • 数据类型:uint32
    • 参数解释:卷积核尺寸
    • 规格限制:默认值1
  • max_displacement:
    • 是否必填:非必填
    • 数据类型:int
    • 参数解释:img1相对于img0的最大偏移量
    • 规格限制:默认值20
  • stride_1:
    • 是否必填:非必填
    • 数据类型:int
    • 参数解释:img0中的像素点遍历的步长
    • 规格限制:默认值1
  • stride_2:
    • 是否必填:非必填
    • 数据类型:int
    • 参数解释:img1中的像素点在img0对应像素点邻域中遍历的步长
    • 规格限制:默认值2

输出

featuremap:
  • 是否必填:必填
  • 数据类型:float16
  • 参数解释:两个输入的featuremap的相似性
  • 规格限制:NCHW