Method _normalization

PATH/core/models/backbones/vit.py:447–450

(x)

    @staticmethod
    def _normalization(x):
        assert len(x.shape) == 4
        x = x.sub(torch.tensor([123.675, 116.280, 103.530]).view(1, 3, 1, 1).cuda()).div(torch.tensor([58.395, 57.120, 57.375]).view(1, 3, 1, 1).cuda())
        return x

    def get_num_layers(self):
        return len(self.blocks)