(self, model, **x)
| 27 | pool_type, in_indices=layer_indices, backbone=backbone) |
| 28 | |
| 29 | def _forward_func(self, model, **x): |
| 30 | backbone_feats = model(mode='extract', **x) |
| 31 | pooling_feats = self.multi_pooling(backbone_feats) |
| 32 | flat_feats = [xx.view(xx.size(0), -1) for xx in pooling_feats] |
| 33 | feat_dict = {'feat{}'.format(i + 1): feat.cpu() \ |
| 34 | for i, feat in enumerate(flat_feats)} |
| 35 | return feat_dict |
| 36 | |
| 37 | def extract(self, model, data_loader, distributed=False): |
| 38 | model.eval() |