init

.ipynb_checkpoints/app-checkpoint.py

@@ -11,6 +11,9 @@ import pytorch_lightning as pl
 from torch.utils.data import DataLoader
 from torchmetrics import Accuracy
 from torchvision import transforms
+from PIL import Image
+import PIL
+
 HF_DATASETS_CACHE="./"
 
 
@@ -118,7 +121,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
             success, frame = capture.read()
 
             if success:
-                cv2.imwrite(f'{class_d}/frame_{frameNr}.jpg', frame)
+                # cv2.imwrite(f'{class_d}/frame_{frameNr}.jpg', frame)
+                frame=frame.save(f'{class_d}/frame_{frameNr}.jpg')
 
             else:
                 break

app.py

@@ -11,6 +11,9 @@ import pytorch_lightning as pl
 from torch.utils.data import DataLoader
 from torchmetrics import Accuracy
 from torchvision import transforms
+from PIL import Image
+import PIL
+
 HF_DATASETS_CACHE="./"
 
 
@@ -118,7 +121,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
             success, frame = capture.read()
 
             if success:
-                cv2.imwrite(f'{class_d}/frame_{frameNr}.jpg', frame)
+                # cv2.imwrite(f'{class_d}/frame_{frameNr}.jpg', frame)
+                frame=frame.save(f'{class_d}/frame_{frameNr}.jpg')
 
             else:
                 break