Python Tensorize Exemples

Exemple #1

    def transforms(self):
        train_transforms = Compose([
            Resize(size=self.size),
            Normalize() if self.normalize else Tensorize()
        ])
        val_transforms = Compose([
            Resize(size=self.size),
            Normalize() if self.normalize else Tensorize()
        ])

        return train_transforms, val_transforms

Exemple #2

Fichier : freiburg_forest.py Projet : ShigemichiMatsuzaki/MSPL

 def transforms(self):
     train_transforms = Compose([
         # RandomScale(scale=self.scale),
         RandomCrop(crop_size=self.size),
         RandomFlip(),
         Normalize() if self.normalize else Tensorize()
     ])
     val_transforms = Compose([
         Resize(size=self.size),
         Normalize() if self.normalize else Tensorize()
     ])
     return train_transforms, val_transforms