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

Method __init__

models/modules/aspp.py:72–77  ·  view source on GitHub ↗
(self, in_channels, planes, kernel_size, padding)

Source from the content-addressed store, hash-verified

70##################### Deformable
71class _ASPPModuleDeformable(nn.Module):
72 def __init__(self, in_channels, planes, kernel_size, padding):
73 super(_ASPPModuleDeformable, self).__init__()
74 self.atrous_conv = DeformableConv2d(in_channels, planes, kernel_size=kernel_size,
75 stride=1, padding=padding, bias=False)
76 self.bn = nn.BatchNorm2d(planes) if config.batch_size > 1 else nn.Identity()
77 self.relu = nn.ReLU(inplace=True)
78
79 def forward(self, x):
80 x = self.atrous_conv(x)

Callers

nothing calls this directly

Calls 2

DeformableConv2dClass · 0.90
__init__Method · 0.45

Tested by

no test coverage detected