【问题标题】:Can I assign the coordinate that convolutional filter works on?我可以分配卷积滤波器工作的坐标吗?
【发布时间】:2018-08-16 14:08:57
【问题描述】:

我不希望conv内核滑动图像中的每个像素,我只需要对我分配的像素进行conv操作。例如,在 224 x 224 的图像上,通常 conv 操作将滑动所有 224 x 224 = 50176 像素。但是,如果我只希望它在 (15,67)、(34, 90)、(143,201).... 上进行转换,我该怎么做? 顺便说一句,我希望这些操作都在 GPU 上运行。 非常感谢!

【问题讨论】:

  • 如果您事先知道您感兴趣的区域,请提取并使用这些补丁,而不是整个图像

标签: tensorflow computer-vision deep-learning caffe pytorch


【解决方案1】:

如果您有内核K(形状为H_k x W_k x C_in x C_out)和图像I(形状为MB x H x W x C_in),并且您有一个位置要应用内核h, w,那么您不需要不需要使用卷积,你可以直接用广播矩阵乘法应用内核:

Out = torch.matmul(I[:, h:h+H_k, w:w+W_k, None, :], K.unsqueeze(0))[:, :, :, 0, :]

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2021-08-04
    • 2019-11-24
    • 1970-01-01
    • 2017-11-22
    • 1970-01-01
    • 2021-01-03
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多