Method
__init__
(self, net_enc, net_dec, crit, deep_sup_scale=None)
Source from the content-addressed store, hash-verified
| 23 | |
| 24 | class SegmentationModule(SegmentationModuleBase): |
| 25 | def __init__(self, net_enc, net_dec, crit, deep_sup_scale=None): |
| 26 | super(SegmentationModule, self).__init__() |
| 27 | self.encoder = net_enc |
| 28 | self.decoder = net_dec |
| 29 | self.crit = crit |
| 30 | self.deep_sup_scale = deep_sup_scale |
| 31 | |
| 32 | def forward(self, feed_dict, *, segSize=None): |
| 33 | if segSize is None: # training |
Callers
nothing calls this directly
Tested by
no test coverage detected