Ubuntu commited on
Commit
3991567
·
1 Parent(s): 2977434
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -19,8 +19,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
19
 
20
  data_dir = Path(str(user_name)+'/train')
21
  transform = transforms.Compose([
22
- transforms.ConvertImageDtype(torch.float32),
23
  transforms.ToTensor(),
 
24
  transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
25
  ])
26
  train_ds = ImageFolder(data_dir, transform=transform)
 
19
 
20
  data_dir = Path(str(user_name)+'/train')
21
  transform = transforms.Compose([
 
22
  transforms.ToTensor(),
23
+ transforms.ConvertImageDtype(torch.float32),
24
  transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
25
  ])
26
  train_ds = ImageFolder(data_dir, transform=transform)
app.py CHANGED
@@ -19,8 +19,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
19
 
20
  data_dir = Path(str(user_name)+'/train')
21
  transform = transforms.Compose([
22
- transforms.ConvertImageDtype(torch.float32),
23
  transforms.ToTensor(),
 
24
  transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
25
  ])
26
  train_ds = ImageFolder(data_dir, transform=transform)
 
19
 
20
  data_dir = Path(str(user_name)+'/train')
21
  transform = transforms.Compose([
 
22
  transforms.ToTensor(),
23
+ transforms.ConvertImageDtype(torch.float32),
24
  transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
25
  ])
26
  train_ds = ImageFolder(data_dir, transform=transform)