a

.ipynb_checkpoints/app-checkpoint.py

@@ -58,11 +58,11 @@ def video_identity(video,user_name,class_name,trainortest,ready):
             transforms.ToTensor(),
             transforms.ConvertImageDtype(torch.float)
             ]) 
-        train_ds = ImageFolder(data_dir)
+        train_ds = ImageFolder(data_dir, transform=transform)
 
         
         test_dir = Path(str(user_name)+'/test')
-        test_ds = ImageFolder(test_dir)        
+        test_ds = ImageFolder(test_dir, transform=transform)        
         
         label2id = {}
         id2label = {}

app.py

@@ -58,11 +58,11 @@ def video_identity(video,user_name,class_name,trainortest,ready):
             transforms.ToTensor(),
             transforms.ConvertImageDtype(torch.float)
             ]) 
-        train_ds = ImageFolder(data_dir)
+        train_ds = ImageFolder(data_dir, transform=transform)
 
         
         test_dir = Path(str(user_name)+'/test')
-        test_ds = ImageFolder(test_dir)        
+        test_ds = ImageFolder(test_dir, transform=transform)        
         
         label2id = {}
         id2label = {}