Merge pull request #50 from NET-BYU/update_video_processing.py

Update video processing.py

demos/video/video_processing.py

@@ -44,6 +44,7 @@
 
 # Create directory for pre-processed videos
 path = "./demos/video/resources/videos/"
+path_processed = "./demos/video/resources/pre-processed/"
 try:
     os.mkdir(path)
 except:
@@ -68,18 +69,23 @@ def normalize(x):
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     # Check if the video has already been processed
-    if not os.path.exists(f"{path}{target[:-4]}.csv"):
-        processing = True
-        logger.info(f"processing {target}")
-    else:
+    if os.path.exists(f"{path_processed}{target[:-4]}.npz"):
         processing = False
         logger.info(f"{target} has already been processed")
+    else:
+        processing = True
+        logger.info(f"processing {target}")
 
     # Iterate through all frames in the video
     video = np.array([])
     while processing:
         (ret, frame) = cap.read()
         if not ret:
+            # Reshape the video and save it
+            video = video.reshape((total_frames, 48, 48))
+            video = video.astype(np.uint8)
+            np.savez_compressed(f"{path_processed}{target[:-4]}.npz", video)
+            logger.info(f"processed {target}")
             break
 
         # fmt: off
@@ -92,12 +98,5 @@ def normalize(x):
         video = np.append(video, graySmall)                    # append the frame to the video
         # fmt: on
 
-    # Reshape the video and save it
-    video = video.reshape((total_frames, 48, 48))
-    video = video.astype(np.uint8)
-    np.savez_compressed(
-        f"./demos/video/resources/pre-processed/{target[:-4]}.npz", video
-    )
-
     # Release the video
     cap.release()