MCPcopy Create free account
hub / github.com/NVIDIA/semantic-segmentation / forward

Method forward

network/ocr_utils.py:149–158  ·  view source on GitHub ↗
(self, feats, proxy_feats)

Source from the content-addressed store, hash-verified

147 )
148
149 def forward(self, feats, proxy_feats):
150 context = self.object_context_block(feats, proxy_feats)
151
152 if cfg.MODEL.OCR_ASPP:
153 aspp = self.aspp(feats)
154 output = self.conv_bn_dropout(torch.cat([context, aspp, feats], 1))
155 else:
156 output = self.conv_bn_dropout(torch.cat([context, feats], 1))
157
158 return output

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected