(self, features)
| 256 | self.mask_forward = mask_forward |
| 257 | |
| 258 | def forward(self, features): |
| 259 | if self.mask_map and self.mask_forward: |
| 260 | features.update({'neck_output': {'mask_features': self.mask_map(features['backbone_output']), |
| 261 | 'multi_scale_features': [features['backbone_output']]}}) |
| 262 | else: |
| 263 | features.update({'neck_output': {'mask_features': None, |
| 264 | 'multi_scale_features': [features['backbone_output']]}}) |
| 265 | return features |
| 266 | |
| 267 | |
| 268 | class ShuffleFPN(SimpleFPN): |