Img Extractor Complete + img

README.md

@@ -1,3 +1,11 @@
 # heatrateestimation
 
-This file contains our work of the heart rate estimation of a person from a video.
+This file contains our work of the heart rate estimation of a person from a video.
+
+## Python
+
+The python will deal with the image extraction and will have to be run each time for new videos.
+
+## Octave
+
+The octave file will work on the bulk of the project to process the data.

frame_extraction.py

@@ -1,4 +1,3 @@
-############
 # frame extraction
 # 
 # Task:
@@ -16,4 +15,21 @@
 # Author: Thomas Périn
 # Date: 17/02/2023
 #
+# Note: check video name when extracting a new one
 ############
+
+import cv2
+
+#Select video by name
+vidcap = cv2.VideoCapture('face_video.mp4')
+success,image = vidcap.read()
+
+#Counter to number the images
+count = 0
+
+#Extraction loop for all images
+while success:
+	cv2.imwrite("img%d.png" % count, image)     # save frame as JPEG file      
+	success,image = vidcap.read()
+	print('Read a new frame: ', success)
+	count += 1