Create frames directory if it doesn't exist

2023-11-15 13:54:16 -05:00 · 2023-11-15 13:54:16 -05:00 · f3c4337fc2
parent c75f298353
commit f3c4337fc2
1 changed files with 5 additions and 0 deletions
--- a/capture.py
+++ b/capture.py
@ -2,10 +2,15 @@ import cv2
 import time
 from PIL import Image
 import numpy as np
+import os

 # Folder
 folder = "frames"

+# Create the frames folder if it doesn't exist
+frames_dir = os.path.join(os.getcwd(), folder)
+os.makedirs(frames_dir, exist_ok=True)
+
 # Initialize the webcam
 cap = cv2.VideoCapture(0)