init

.ipynb_checkpoints/app-checkpoint.py

@@ -18,11 +18,14 @@ def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
 
         data_dir = Path(str(user_name)+'/train')
-        train_ds = ImageFolder(data_dir)
+        transform = transforms.Compose([
+            transforms.ToTensor(),
+            ]) 
+        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

@@ -18,11 +18,14 @@ def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
 
         data_dir = Path(str(user_name)+'/train')
-        train_ds = ImageFolder(data_dir)
+        transform = transforms.Compose([
+            transforms.ToTensor(),
+            ]) 
+        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 = {}