Spaces:
Runtime error
Runtime error
Ubuntu
commited on
Commit
·
59eb24d
1
Parent(s):
5feff63
init
Browse files- .ipynb_checkpoints/app-checkpoint.py +2 -3
- app.py +2 -3
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -76,8 +76,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 76 |
|
| 77 |
|
| 78 |
|
| 79 |
-
train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=
|
| 80 |
-
test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=
|
| 81 |
|
| 82 |
|
| 83 |
for name, param in model.named_parameters():
|
|
@@ -101,7 +101,6 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 101 |
|
| 102 |
else:
|
| 103 |
capture = cv2.VideoCapture(video)
|
| 104 |
-
!sysctl kernel.shmmax
|
| 105 |
user_d=str(user_name)+'/'+str(trainortest)
|
| 106 |
class_d=str(user_name)+'/'+str(trainortest)+'/'+str(class_name)
|
| 107 |
if not os.path.exists(user_d):
|
|
|
|
| 76 |
|
| 77 |
|
| 78 |
|
| 79 |
+
train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=0, shuffle=True)
|
| 80 |
+
test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=0)
|
| 81 |
|
| 82 |
|
| 83 |
for name, param in model.named_parameters():
|
|
|
|
| 101 |
|
| 102 |
else:
|
| 103 |
capture = cv2.VideoCapture(video)
|
|
|
|
| 104 |
user_d=str(user_name)+'/'+str(trainortest)
|
| 105 |
class_d=str(user_name)+'/'+str(trainortest)+'/'+str(class_name)
|
| 106 |
if not os.path.exists(user_d):
|
app.py
CHANGED
|
@@ -76,8 +76,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 76 |
|
| 77 |
|
| 78 |
|
| 79 |
-
train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=
|
| 80 |
-
test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=
|
| 81 |
|
| 82 |
|
| 83 |
for name, param in model.named_parameters():
|
|
@@ -101,7 +101,6 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 101 |
|
| 102 |
else:
|
| 103 |
capture = cv2.VideoCapture(video)
|
| 104 |
-
!sysctl kernel.shmmax
|
| 105 |
user_d=str(user_name)+'/'+str(trainortest)
|
| 106 |
class_d=str(user_name)+'/'+str(trainortest)+'/'+str(class_name)
|
| 107 |
if not os.path.exists(user_d):
|
|
|
|
| 76 |
|
| 77 |
|
| 78 |
|
| 79 |
+
train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=0, shuffle=True)
|
| 80 |
+
test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=0)
|
| 81 |
|
| 82 |
|
| 83 |
for name, param in model.named_parameters():
|
|
|
|
| 101 |
|
| 102 |
else:
|
| 103 |
capture = cv2.VideoCapture(video)
|
|
|
|
| 104 |
user_d=str(user_name)+'/'+str(trainortest)
|
| 105 |
class_d=str(user_name)+'/'+str(trainortest)+'/'+str(class_name)
|
| 106 |
if not os.path.exists(user_d):
|