fix(reidmodel): torch.squeeze removes batch dim for batch (+3 more)

[andrewwhitecdw]

Small fixes in reIDmodel.py:

fix: torch.squeeze removes batch dim for batch size 1

Fix: Replace:

    def forward(self, x):
        x = self.model.features(x)
        x = torch.squeeze(x)
        x = self.classifier(x)
        return x

with:

    def forward(self, x):
        x = self.model.features(x)
        x = x.view(x.size(0), -1)
        x = self.classifier(x)
        return x

fix: torch.squeeze removes batch dim when batch size is 1

Fix: Replace:

        x = torch.cat((x0,x1),1)
        x = torch.squeeze(x)
        x = self.classifier(x)
        return x

with:

        x = torch.cat((x0,x1),1)
        x = x.view(x.size(0), -1)
        x = self.classifier(x)
        return x

fix: PCB part classifier passes bool as dropout rate

Fix: Replace:

            setattr(self, name, ClassBlock(2048, class_num, True, False, 256))

with:

            setattr(self, name, ClassBlock(2048, class_num, droprate=0.5, relu=False, num_bottleneck=256))

fix: ft_net_dense fails on batch size 1 due to torch.squeeze

Fix: Replace:

        x = self.model.features(x)
        x = torch.squeeze(x)
        x = self.classifier(x)

with:

        x = self.model.features(x)
        x = x.view(x.size(0), -1)
        x = self.classifier(x)

Files changed

• reIDmodel.py