如何在TensorFlow中为两个图像编写自定义conv2d层

我的研究要求我制作一个自定义的卷积层。假设我有一个包含两个图像[W,H,2,C]和一个内核[w,w,2,c]的输入。我的目标是解耦每个图像中的内核运动。

也就是说,假设内核位于左上角。在第一步中,我希望内核在图像1中向左移动一步,但仍保持在图像2中的相同位置。然后重复相同的操作,直到内核未完成图像1中的第一行。图1中的内核返回到初始位置,而图2中的内核向右移动了一步。重复相同的算法,直到图像2中的内核未完成第一行。然后将两个内核下移一个步骤,然后重复整个过程。进行同样的操作,直到内核没有清除整个图像为止。

在TensorFlow中实现这种算法有什么办法吗?

zsandy 回答:如何在TensorFlow中为两个图像编写自定义conv2d层

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3140058.html

大家都在问