(self, features)
| 519 | self.fpns = nn.ModuleList([fpn1]) |
| 520 | |
| 521 | def forward(self, features): |
| 522 | x = features['backbone_output'] |
| 523 | |
| 524 | out = [op(x) for op in self.fpns][::-1] # [r4, r3, r2, r1] |
| 525 | multi_scale_features = [] |
| 526 | for o in out: |
| 527 | multi_scale_features.append(o) |
| 528 | |
| 529 | features.update({'neck_output': multi_scale_features}) |
| 530 | |
| 531 | return features |
| 532 | |
| 533 | |
| 534 | class PedDetSimpleResNetFPN(nn.Module): |