init

.ipynb_checkpoints/app-checkpoint.py

@@ -23,8 +23,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
         data_dir = Path(str(user_name)+'/train')
         transform = transforms.Compose([
             transforms.ToTensor(),
-            transforms.ConvertImageDtype(torch.float32),
-            transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
+            transforms.ConvertImageDtype(torch.float)
             ]) 
         train_ds = ImageFolder(data_dir, transform=transform)
 

app.py

@@ -23,8 +23,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
         data_dir = Path(str(user_name)+'/train')
         transform = transforms.Compose([
             transforms.ToTensor(),
-            transforms.ConvertImageDtype(torch.float32),
-            transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
+            transforms.ConvertImageDtype(torch.float)
             ]) 
         train_ds = ImageFolder(data_dir, transform=transform)