Create frames folder if it doesn't exist

2023-11-15 13:46:10 -05:00 · 2023-11-15 13:46:10 -05:00 · 61e458e95b
parent feeb0cfbd2
commit 61e458e95b
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)