(self, fmap1, fmap2, num_levels=4, radius=4)
| 61 | |
| 62 | class AlternateCorrBlock: |
| 63 | def __init__(self, fmap1, fmap2, num_levels=4, radius=4): |
| 64 | self.num_levels = num_levels |
| 65 | self.radius = radius |
| 66 | |
| 67 | self.pyramid = [(fmap1, fmap2)] |
| 68 | for i in range(self.num_levels): |
| 69 | fmap1 = F.avg_pool2d(fmap1, 2, stride=2) |
| 70 | fmap2 = F.avg_pool2d(fmap2, 2, stride=2) |
| 71 | self.pyramid.append((fmap1, fmap2)) |
| 72 | |
| 73 | def __call__(self, coords): |
| 74 | coords = coords.permute(0, 2, 3, 1) |
nothing calls this directly
no outgoing calls
no test coverage detected