MCPcopy Create free account
hub / github.com/MaureenZOU/TSAM / forward

Method forward

src/model/modules.py:267–286  ·  view source on GitHub ↗
(self, inp, flows=None)

Source from the content-addressed store, hash-verified

265 nn.init.constant_(module.conv.featureConv.layer.bias, 0)
266
267 def forward(self, inp, flows=None):
268 c1, c2, c4, c8 = inp
269
270 c2 = self.conv_c2(c2, flows)
271 c4 = self.conv_c4(c4, flows)
272
273 c4 = interpolate(c4, scale_factor=2, mode="nearest")
274 c2 = interpolate(c2, scale_factor=4, mode="nearest")
275 c1 = interpolate(c1, scale_factor=4, mode="nearest")
276
277 d1 = self.deconv1(c8, flows)
278 d1 = d1 + c4
279 c9 = self.conv9(d1, flows)
280 d2 = self.deconv2(c9, flows)
281 d2 = d2 + c2
282 c10 = self.conv10(d2, flows)
283 d3 = self.deconv3(c10, flows)
284 d3 = d3 + c1
285 c11 = self.conv11(d3, flows)
286 return c11

Callers

nothing calls this directly

Calls 1

interpolateFunction · 0.90

Tested by

no test coverage detected