init

.ipynb_checkpoints/app-checkpoint.py

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

app.py

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