fingerprinted now should work correctly

2020-05-20 13:29:16 +02:00
parent 196b1af7ae
commit b87a56e8c6
3 changed files with 4 additions and 2 deletions
--- a/utils/transforms.py
+++ b/utils/transforms.py
@ -4,5 +4,6 @@ from torchvision.transforms import ToTensor as TorchVisionToTensor
 class ToTensor(TorchVisionToTensor):

    def __call__(self, pic):
+        # Make it float .float() == 32bit
        tensor = super(ToTensor, self).__call__(pic).float()
        return tensor