modular structure complete

01-DC · Feb 26, 2022 · d597aec · d597aec
1 parent 9cfd58c
commit d597aec
Show file tree

Hide file tree

Showing 59 changed files with 40 additions and 2,838 deletions.
diff --git a/ExtractedSlides/Frame1.jpg b/ExtractedSlides/Frame1.jpg
diff --git a/ExtractedSlides/Frame10.jpg b/ExtractedSlides/Frame10.jpg
diff --git a/ExtractedSlides/Frame11.jpg b/ExtractedSlides/Frame11.jpg
diff --git a/ExtractedSlides/Frame12.jpg b/ExtractedSlides/Frame12.jpg
diff --git a/ExtractedSlides/Frame13.jpg b/ExtractedSlides/Frame13.jpg
diff --git a/ExtractedSlides/Frame14.jpg b/ExtractedSlides/Frame14.jpg
diff --git a/ExtractedSlides/Frame15.jpg b/ExtractedSlides/Frame15.jpg
diff --git a/ExtractedSlides/Frame16.jpg b/ExtractedSlides/Frame16.jpg
diff --git a/ExtractedSlides/Frame17.jpg b/ExtractedSlides/Frame17.jpg
diff --git a/ExtractedSlides/Frame18.jpg b/ExtractedSlides/Frame18.jpg
diff --git a/ExtractedSlides/Frame19.jpg b/ExtractedSlides/Frame19.jpg
diff --git a/ExtractedSlides/Frame2.jpg b/ExtractedSlides/Frame2.jpg
diff --git a/ExtractedSlides/Frame20.jpg b/ExtractedSlides/Frame20.jpg
diff --git a/ExtractedSlides/Frame21.jpg b/ExtractedSlides/Frame21.jpg
diff --git a/ExtractedSlides/Frame22.jpg b/ExtractedSlides/Frame22.jpg
diff --git a/ExtractedSlides/Frame23.jpg b/ExtractedSlides/Frame23.jpg
diff --git a/ExtractedSlides/Frame24.jpg b/ExtractedSlides/Frame24.jpg
diff --git a/ExtractedSlides/Frame25.jpg b/ExtractedSlides/Frame25.jpg
diff --git a/ExtractedSlides/Frame26.jpg b/ExtractedSlides/Frame26.jpg
diff --git a/ExtractedSlides/Frame27.jpg b/ExtractedSlides/Frame27.jpg
diff --git a/ExtractedSlides/Frame28.jpg b/ExtractedSlides/Frame28.jpg
diff --git a/ExtractedSlides/Frame29.jpg b/ExtractedSlides/Frame29.jpg
diff --git a/ExtractedSlides/Frame3.jpg b/ExtractedSlides/Frame3.jpg
diff --git a/ExtractedSlides/Frame30.jpg b/ExtractedSlides/Frame30.jpg
diff --git a/ExtractedSlides/Frame31.jpg b/ExtractedSlides/Frame31.jpg
diff --git a/ExtractedSlides/Frame32.jpg b/ExtractedSlides/Frame32.jpg
diff --git a/ExtractedSlides/Frame33.jpg b/ExtractedSlides/Frame33.jpg
diff --git a/ExtractedSlides/Frame34.jpg b/ExtractedSlides/Frame34.jpg
diff --git a/ExtractedSlides/Frame35.jpg b/ExtractedSlides/Frame35.jpg
diff --git a/ExtractedSlides/Frame36.jpg b/ExtractedSlides/Frame36.jpg
diff --git a/ExtractedSlides/Frame37.jpg b/ExtractedSlides/Frame37.jpg
diff --git a/ExtractedSlides/Frame38.jpg b/ExtractedSlides/Frame38.jpg
diff --git a/ExtractedSlides/Frame39.jpg b/ExtractedSlides/Frame39.jpg
diff --git a/ExtractedSlides/Frame4.jpg b/ExtractedSlides/Frame4.jpg
diff --git a/ExtractedSlides/Frame40.jpg b/ExtractedSlides/Frame40.jpg
diff --git a/ExtractedSlides/Frame41.jpg b/ExtractedSlides/Frame41.jpg
diff --git a/ExtractedSlides/Frame42.jpg b/ExtractedSlides/Frame42.jpg
diff --git a/ExtractedSlides/Frame43.jpg b/ExtractedSlides/Frame43.jpg
diff --git a/ExtractedSlides/Frame44.jpg b/ExtractedSlides/Frame44.jpg
diff --git a/ExtractedSlides/Frame45.jpg b/ExtractedSlides/Frame45.jpg
diff --git a/ExtractedSlides/Frame46.jpg b/ExtractedSlides/Frame46.jpg
diff --git a/ExtractedSlides/Frame47.jpg b/ExtractedSlides/Frame47.jpg
diff --git a/ExtractedSlides/Frame48.jpg b/ExtractedSlides/Frame48.jpg
diff --git a/ExtractedSlides/Frame49.jpg b/ExtractedSlides/Frame49.jpg
diff --git a/ExtractedSlides/Frame5.jpg b/ExtractedSlides/Frame5.jpg
diff --git a/ExtractedSlides/Frame50.jpg b/ExtractedSlides/Frame50.jpg
diff --git a/ExtractedSlides/Frame51.jpg b/ExtractedSlides/Frame51.jpg
diff --git a/ExtractedSlides/Frame52.jpg b/ExtractedSlides/Frame52.jpg
diff --git a/ExtractedSlides/Frame53.jpg b/ExtractedSlides/Frame53.jpg
diff --git a/ExtractedSlides/Frame6.jpg b/ExtractedSlides/Frame6.jpg
diff --git a/ExtractedSlides/Frame7.jpg b/ExtractedSlides/Frame7.jpg
diff --git a/ExtractedSlides/Frame8.jpg b/ExtractedSlides/Frame8.jpg
diff --git a/ExtractedSlides/Frame9.jpg b/ExtractedSlides/Frame9.jpg
diff --git a/frameExtractor.py b/frameExtractor.py
@@ -1,22 +1,25 @@
 import cv2
 import os
 
+def frameExtractor(PATH, VIDEO_PATH, selectedFrames):
+    vid= cv2.VideoCapture(VIDEO_PATH)
 
-def frameExtractor():
     if not os.path.exists(PATH):
         os.makedirs(PATH)
 
-    for framePos in selectedFrames:
+    for i, framePos in enumerate(selectedFrames):
         vid.set(cv2.CAP_PROP_POS_FRAMES, framePos-1)
         success, frame= vid.read()
 
         if not success:
             print('Video END or Video File corrupted.')
             break
 
-        print("Selected Frame:", keyframe_count, ' @ ', vid.get(cv2.CAP_PROP_POS_MSEC)/1000, 'seconds')
-        cv2.imwrite(os.path.join(PATH, 'Frame{}.jpg'.format(keyframe_count)), frame)
-        keyframe_count+=1
+        print('Selected Frame: {} @ {} seconds'.format(i+1, vid.get(cv2.CAP_PROP_POS_MSEC)/1000))
+
+        cv2.imwrite(os.path.join(PATH, 'Frame{}.jpg'.format(i+1)), frame)
 
+    vid.release()
 
-    vid.release()
+if __name__ == '__main__':
+    pass
diff --git a/frameSelector.py b/frameSelector.py
@@ -2,25 +2,25 @@
 from matplotlib import pyplot as plt
 
 def frameSelector() -> list[float]:
-    frameHashValue, frameNumber= [], []
-    with open('frame_hash_data.txt', 'r') as f:
+    frameHashValue, framePosition= [], []
+    with open('frameHashData.txt', 'r') as f:
         for line in f:
             x, y= line.strip().split()
             frameHashValue.append(int(x))
-            frameNumber.append(float(y))
+            framePosition.append(float(y))
 
     ch= input('Plot of analysed values? (Y/N): ')
     if ch == 'Y':
-        plt.plot(frameNumber, frameHashValue)
+        plt.plot(framePosition, frameHashValue)
         plt.title('Analysed frame difference values')
         plt.show()
 
     thresh= int(input("Enter Threshold value (recommended between 350 to 400): "))
 
-    selectedFrames= []
-    for i, val in enumerate(frameHashValue):
-        if val >= thresh:
-            selectedFrames.append(frameNumber[i])
+    selectedFrames= [framePosition[i] for i, val in enumerate(frameHashValue) if val>=thresh]
+    # for i, val in enumerate(frameHashValue):
+    #     if val >= thresh:
+    #         selectedFrames.append(framePosition[i])
 
     return selectedFrames