(self, backbone, head)
| 581 | concatenated features. |
| 582 | """ |
| 583 | def __init__(self, backbone, head): |
| 584 | super(MultiCropWrapper, self).__init__() |
| 585 | # disable layers dedicated to ImageNet labels classification |
| 586 | backbone.fc, backbone.head = nn.Identity(), nn.Identity() |
| 587 | self.backbone = backbone |
| 588 | self.head = head |
| 589 | |
| 590 | def forward(self, x): |
| 591 | # convert to list |