使用数据库
import cv2
import os
import shutil
参数
video_name = "001.mp4" # 待处理视频
para = 10 # 每隔10帧提取一次
c = 0 # 参数
创建保存图片的文件夹
def make_dir(filepath):
# 如果文件夹存在就删除,不存在就创建
if not os.path.exists(filepath):
os.mkdir(filepath)
else:
shutil.rmtree(filepath)
os.mkdir(filepath)
video = cv2.VideoCapture(video_name)
rval, frame = video.read()
make_dir("./damage_picture")
开始截取特定帧,并将图片文件名的字符长度设定为一致
while rval:
rval, frame = video.read()
if c % para == 0:
img_name = format(c, "06d") # 图片的名称都为 000001.jpg、000002.jpg等类型,06表示位数为6位
cv2.imwrite("damage_picture/" + str(img_name) + ".jpg", frame)
c += 1
cv2.waitKey()
video.release()