python遍历目录下的所有目录和文件，并用opencv从mp4文件中抽帧得到图片

做深度学习训练的时候，需要从mp4视频文件中抽帧得到图片数据，其中mp4文件是保存在一个大文件夹下的不同子文件夹下面的，用下面的python脚本实现

import cv2
import os"""根据传入的目录参数，得到该目录所有子文件夹下的所有的mp4文件"""
def get_mp4path(main_dir):list_mp4 = []for root, dirs, files in os.walk(main_dir):for file in files:if file.endswith(".mp4"):  #只要mp4文件，#print(os.path.join(root, file))list_mp4.append(os.path.join(root, file))return list_mp4 #得到所有的mp4文件的列表(包含路径)"""根据传入的目录地址和MP4文件的名字，从视频文件中抽帧得到图片数据并保存"""
def get_image(path):print(path)#调试，记得删除mp4_path, mp4_name = os.path.split(path)  #分别得到目录和文件名