视频转图片
import cv2
import os
img_count = 1
def save_frames(video_path, output_folder, interval):
if not os.path.exists(output_folder):
os.makedirs(output_folder)
cap = cv2.VideoCapture(video_path)
frame_count = 0
global img_count
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
frame_count += 1
if frame_count % (interval * cap.get(cv2.CAP_PROP_FPS)) == 0:
img_name = f"20240524_{img_count}.jpg"
img_path = os.path.join(output_folder, img_name)
cv2.imwrite(img_path, frame)
print('file saved:', img_path)
print(img_count)
img_count += 1
cap.release()
input_folder = "path"
output_folder = "path"
for filename in os.listdir(input_folder):
if filename.endswith(".dav"):
video_path = os.path.join(input_folder, filename)
save_frames(video_path, output_folder, 1)
双目视频,提取单目
import cv2
import os
img_count = 1
def extract_left_eye_frame(video_path, output_folder):
global img_count
cap = cv2.VideoCapture(video_path)
frame_rate = cap.get(cv2.CAP_PROP_FPS)
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
half_width = frame_width // 2
last_saved_time = 0.0
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
current_time = cap.get(cv2.CAP_PROP_POS_MSEC) / 1000.0
if current_time - last_saved_time >= 1.0:
left_eye_frame = frame[:, :half_width]
img_name = f"20240422xiaomi_{img_count}.jpg"
img_path = os.path.join(output_folder, img_name)
cv2.imwrite(img_path, left_eye_frame)
print(f"Saved: {img_path}")
img_count += 1
last_saved_time = current_time
cap.release()
input_folder = "path"
output_folder = "path"
if not os.path.exists(output_folder):
os.makedirs(output_folder)
for filename in os.listdir(input_folder):
if filename.endswith(".mp4"):
print(f"Processing: {filename}")
video_path = os.path.join(input_folder, filename)
extract_left_eye_frame(video_path, output_folder)