Spaces:
Runtime error
Runtime error
Ubuntu
commited on
Commit
·
033a668
1
Parent(s):
e25be51
init
Browse files- .ipynb_checkpoints/app-checkpoint.py +4 -4
- app.py +4 -4
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -21,10 +21,10 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 21 |
if ready=='yes':
|
| 22 |
|
| 23 |
data_dir = Path(str(user_name)+'/train')
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
train_ds = ImageFolder(data_dir)
|
| 29 |
|
| 30 |
|
|
|
|
| 21 |
if ready=='yes':
|
| 22 |
|
| 23 |
data_dir = Path(str(user_name)+'/train')
|
| 24 |
+
transform = transforms.Compose([
|
| 25 |
+
transforms.ToTensor(),
|
| 26 |
+
transforms.ConvertImageDtype(torch.float)
|
| 27 |
+
])
|
| 28 |
train_ds = ImageFolder(data_dir)
|
| 29 |
|
| 30 |
|
app.py
CHANGED
|
@@ -21,10 +21,10 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 21 |
if ready=='yes':
|
| 22 |
|
| 23 |
data_dir = Path(str(user_name)+'/train')
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
train_ds = ImageFolder(data_dir)
|
| 29 |
|
| 30 |
|
|
|
|
| 21 |
if ready=='yes':
|
| 22 |
|
| 23 |
data_dir = Path(str(user_name)+'/train')
|
| 24 |
+
transform = transforms.Compose([
|
| 25 |
+
transforms.ToTensor(),
|
| 26 |
+
transforms.ConvertImageDtype(torch.float)
|
| 27 |
+
])
|
| 28 |
train_ds = ImageFolder(data_dir)
|
| 29 |
|
| 30 |
|