import cv2
import glob
import os
import gdalTools
def Video2Pic(videoPath, imgPath):
cap = cv2.VideoCapture(videoPath)
suc = cap.isOpened()
frame_count = 0
baseName = os.path.basename(videoPath).split(".")[0]
while suc:
frame_count += 1
suc, frame = cap.read()
cv2.imwrite(os.path.join(imgPath, f"{baseName}_{str(frame_count).zfill(4)}.png"), frame)
cv2.waitKey(1)
cap.release()
print("视频转图片结束!")
if __name__ == '__main__':
videoPathList = glob.glob(r"D:\2022\3\medicalSeg\video\kidney\*.avi")
imgPath = "../images"
gdalTools.mkdir(imgPath)
for videoPath in videoPathList:
Video2Pic(videoPath, imgPath)