astatin3/vid-to-3d
1
0
Fork 0
mirror of https://github.com/Astatin3/vid-to-3d.git synced 2026-06-09 00:28:02 -06:00
Code Issues Packages Projects Releases Wiki Activity

Some random changes

Browse Source
This commit is contained in:
Astatin3
2024-08-02 19:39:19 -06:00
parent 2e3969bec3
commit b9c8c79e00
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+10 -5
View File
@@ -16,22 +16,23 @@ def process_video(video_path, min_similarity, output_folder):
prev_frame = None prev_frame = None
kp1, ds1 = None, None kp1, ds1 = None, None
i = 0 i = 0
while cap.isOpened(): while cap.isOpened():
ret, frame = cap.read() ret, frame = cap.read()
if not ret: break if not ret: break
i += 1 i += 1
print(f"Frame {i}: ", end='') print(f"Frame {i}: ", end='')
if prev_frame is None: if prev_frame is None:
prev_frame = frame kp1, des1 = orb.detectAndCompute(frame, None)
kp1, des1 = orb.detectAndCompute(prev_frame, None) cv2.imwrite(os.path.join(output_folder, os.path.basename(video_path)+"-"+str(i)+".png"), frame)
cv2.imwrite(os.path.join(output_folder, os.path.basename(video_path)+"-"+str(i)+".png"), prev_frame)
print("Initial frame") print("Initial frame")
prev_frame = frame
continue continue
cv2.imshow('Frame', frame) cv2.imshow('Frame', frame)
@@ -61,12 +62,16 @@ def process_video(video_path, min_similarity, output_folder):
orb = cv2.ORB_create() orb = cv2.ORB_create()
bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True) bf = cv2.BFMatcher(cv2.NORM_L1, crossCheck=True)
def orb_similarity(kp1, des1, img2): def orb_similarity(kp1, des1, img2):
kp2, des2 = orb.detectAndCompute(img2, None) kp2, des2 = orb.detectAndCompute(img2, None)
matches = bf.match(des1, des2) matches = bf.match(des1, des2)
# try:
# matches = bf.match(des1, des2)
# except:
# return 0.0
similarity = len(matches) / min(len(kp1), len(kp2)) similarity = len(matches) / min(len(kp1), len(kp2))
return similarity return similarity