(self, x)
| 89 | ) |
| 90 | |
| 91 | def forward(self, x): |
| 92 | x = self.conv(x) |
| 93 | x = self.layer1(x) |
| 94 | x = self.layer2(x) |
| 95 | x = self.layer3(x) |
| 96 | x = self.layer4(x) |
| 97 | x = self.avgpool(x) |
| 98 | x = x.view(x.size(0), -1) |
| 99 | # B x 128 |
| 100 | if self.reid: |
| 101 | x = x.div(x.norm(p=2, dim=1, keepdim=True)) |
| 102 | return x |
| 103 | # classifier |
| 104 | x = self.classifier(x) |
| 105 | return x |
| 106 | |
| 107 | |
| 108 | class Extractor(object): |