Templates finalized

2020-05-17 22:05:20 +02:00
parent 398504aee6
commit fc93f71608
20 changed files with 459 additions and 166 deletions
--- a/utils/transforms.py
+++ b/utils/transforms.py
@ -1,8 +1,8 @@
-from torchvision.transforms import ToTensor as TorchvisionToTensor
+from torchvision.transforms import ToTensor as TorchVisionToTensor


-class ToTensor(TorchvisionToTensor):
+class ToTensor(TorchVisionToTensor):

    def __call__(self, pic):
        tensor = super(ToTensor, self).__call__(pic).float()
-        return tensor
+        return tensor