MCPcopy Create free account
hub / github.com/ZhengPeng7/BiRefNet / forward

Method forward

models/modules/aspp.py:54–67  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

52 self.dropout = nn.Dropout(0.5)
53
54 def forward(self, x):
55 x1 = self.aspp1(x)
56 x2 = self.aspp2(x)
57 x3 = self.aspp3(x)
58 x4 = self.aspp4(x)
59 x5 = self.global_avg_pool(x)
60 x5 = F.interpolate(x5, size=x1.size()[2:], mode='bilinear', align_corners=True)
61 x = torch.cat((x1, x2, x3, x4, x5), dim=1)
62
63 x = self.conv1(x)
64 x = self.bn1(x)
65 x = self.relu(x)
66
67 return self.dropout(x)
68
69
70##################### Deformable

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected