import cv2
import os
video_path = '/home/ly/Desktop/视频标注/camera.avi' #视频路径
picSavePath = '/home/ly/Desktop/视频标注/img2' #自定义图片保存路径
cap = cv2.VideoCapture(video_path) #读取视频
TOTAL_FRAME = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) #获取视频总帧数
print(TOTAL_FRAME)
'''如果目录picSavePath不存在就创建这个目录'''
if not os.path.isdir(picSavePath):
os.makedirs(picSavePath)
saveDir = picSavePath
print(saveDir)
counter = 0 #计算帧数
if cap.isOpened():
while True:
ret, frame = cap.read() #读取视频帧
if ret == False: # 判断是否读取成功
break
counter += 1 #命名编号+1
imgname = "carLine%s.jpg" % str(counter) #图片命名
path = os.path.join(saveDir, imgname) #图片保存路径
print(path)
cv2.imwrite(path, frame) #写入图片
cap.release() #释放资源
print("finish!") #转换结束