简单代码实现视频转图片_py
目录
1.安装OpenCV
环境要求
安装命令
验证安装
2. OpenCV用法
3.实现程序
博主最近在研究深度学习，需要收集数据集进行处理，但一张张拍照真是太麻烦了
就想着，哎，能不能写一个程序，把视频转成图片不就行了？
正好，博主知道Python中就有一个库可以做到哎。
import cv2
是的，就是咱们强大的cv2库了
这是一个处理视频很好用的库，既然如此，我们就来看看我们需要的用法吧。
1.安装OpenCV
环境要求
确保你的Python版本至少为3.6以上，推荐使用Python 3.8。
安装命令
在你的Python环境中安装OpenCV非常简单，使用pip即可
pip install opencv-python
验证安装
安装完成后，可以通过以下Python代码来验证cv2模块是否正确安装
import cv2
print(cv2.__version__)
2. OpenCV用法
大家直接去官方文档里看吧，这边就不具体介绍了，因为这个库真是太强大了。
OpenCV官方文档 OpenCV中文文档
当然，还是推荐大家（如果英语好的话）使用官方文档，毕竟别人翻译的终究不准确。
3.实现程序
首先我们需要创建两个文件夹目录来存储我们的图片与视频
import cv2
#视频路径
video_path = r'D:video\1.mp4'
#1.mp4就是我们的视频
#图片保存路径
out_path = 'D:images/'接下来，我们就需要对视频进行读取，我们先创建一个实例化对象
videocapture = cv2.VideoCapture(video_path) #创建一个实例化对象（捕捉视频）
success, frame = videocapture.read() #输出success表示是否读到帧
print("success is")
print(success)这边我们运行一下代码，出现这段文件就说明我们视频读取成功了
这时候我们就需要设置一下读取视频的参数了，代码中注释的很清楚，就不过多解释
is_all_frame = True #是否取所有帧
sta_frame = 1 # 起始帧
end_frame = 1000 #结束帧
time_interval = 8 #时间间隔,多少帧存一次根据我们的参数可以知道，我们每8帧就需要存储一次，自然就需要一个循环来不停的存储我们的
图片。有了循环，自然就得有循环结束的条件。接下来，看代码。
i = 0
j = 0 # 为了编号，防止图片编号跳着来 1,8,16....
while success:
i += 1
if i % time_interval == 0:
if not is_all_frame:
if sta_frame <= i <= end_frame: #i在启始帧和结束帧之间运行循环
j += 1
print("save frame", j)
save_image(out_path, frame, j)
elif i > end_frame:
break
else:
j += 1
print("save frame", j)
save_image(out_path, frame, j)
# 每次调用读取一帧
success, frame = videocapture.read()
# save_image(out_path, frame, j)保存图片的函数，暂时可以不看，下面会讲
# 变量j也可以暂时不看循环有了，我们就要写我们需要用的具体函数了。
这边我们定义一个save_image（） 函数，设置几个参数，分别存放我们的目录，每一帧图片，图
片的编号 j 。
为什么我们需要编号呢？
如果我们直接用frame作为编号，就会出现编号跳跃的情况，8,16,24...
这样看着会很难受，引入 j 只是为了让我们看的方便些。
接下来，看代码
def save_image(addr, image, num): #目录，每一帧，编号
address = addr + 'img_' + str(num) + '.jpg' #路径以及文件保存的名字
print(address) #输出路径，为了直观一点
cv2.imwrite(address, image) #转换图片的主要程序这样，我们就可以完成一个简单的视频转图片的程序了。
4.完整代码
import cv2
#这边的路径是博主自己进行深度学习的目录，改为你们自己的就可以了
#视频路径
video_path = r'D:\yolo\yolov5-6.0\labelimg\video\1.mp4'
#图片路径
out_path = 'D:/yolo/yolov5-6.0/labelimg/images/'
# 保存图片函数
def save_image(addr, image, num): #目录，图片，编号
address = addr + 'img_' + str(num) + '.jpg'
print(address)
cv2.imwrite(address, image)
is_all_frame = True #是否取所有帧
sta_frame = 1 # 起始帧
end_frame = 1000 #结束帧
# 这边可以根据需要自定义
# 拍摄视频 1s有60帧 （具体看自己相机设置）
time_interval = 8 #时间间隔,多少帧存一次
videocapture = cv2.VideoCapture(video_path) #创建一个实例化对象（捕捉视频）
success, frame = videocapture.read() #success表示是否读到帧
print("success is")
print(success)
i = 0
j = 0 # 为了编号，防止图片编号跳着来 1,8,16....
while success:
i += 1
if i % time_interval == 0:
if not is_all_frame:
if sta_frame <= i <= end_frame:
j += 1
print("save frame", j)
save_image(out_path, frame, j)
elif i > end_frame:
break
else:
j += 1
print("save frame", j)
save_image(out_path, frame, j)
# 每次调用读取一帧
success, frame = videocapture.read()
