接下來安裝pytorch?
Previous PyTorch Versions
視頻抽幀?
import cv2def extract_frames(video_path, output_folder, frame_rate=1):"""從視頻中抽取幀。:param video_path: 視頻文件的路徑:param output_folder: 存儲幀的文件夾路徑:param frame_rate: 抽取的幀率,默認為1(每秒1幀)"""# 打開視頻文件cap = cv2.VideoCapture(video_path)if not cap.isOpened():print("Error: Could not open video.")return# 獲取視頻的幀率,這對于控制抽取幀的間隔很有用fps = cap.get(cv2.CAP_PROP_FPS)frame_interval = int(fps / frame_rate) # 計算間隔幀數frame_count = 0frame_index = 0# 循環讀取幀直到視頻結束while True:ret, frame = cap.read()if not ret:break # 如果讀取失敗,則退出循環# 根據幀率間隔抽取幀if frame_count % frame_interval == 0:frame_filename = f"{output_folder}/frame_{frame_index:05d}.jpg"cv2.imwrite(frame_filename, frame) # 保存幀print(f"Saved {frame_filename}")frame_index += 1frame_count += 1# 釋放視頻捕獲對象cap.release()print("Extraction complete.")# 使用示例
video_path = '20035.MP4' # 視頻文件路徑
output_folder = 'imgs' # 輸出文件夾路徑
extract_frames(video_path, output_folder, frame_rate=2) # 每秒抽取2幀