Class DinoImageNormalizeTransform

data/transforms_vggt.py:27–45

class DinoImageNormalizeTransform:
    def __init__(
        self, 
        mode='crop',
        target_size=518
    ):
        self.mode = mode
        self.target_size = target_size

        # self.to_tensor_transform = transforms.ToTensor()
        self.normalize_transform = transforms.Normalize(mean=_RESNET_MEAN, std=_RESNET_STD, inplace=True)

    def __call__(self, img, img_num=1):
       
        # img = self.to_tensor_transform(img)  # Convert to tensor (0, 1)
        img = self.normalize_transform(img)

        shape = (img.shape[1], img.shape[2]) 
        return img
    
class DinoImageTransform:
    def __init__(