注释:图片名字以数字顺序为序列,0.jpg,1.jpg…
1、图片转成视频
import cv2
import os
# 读取文件夹的路径,这里处理多个文件夹
IMG_DIR_1 = '/media/gooddz/xiaoliu/code/LSTR-main/晴天1检测结果/'
IMG_DIR_2 = '/media/gooddz/xiaoliu/code/LSTR-main/晴天2检测结果/'
IMG_DIR_3 = '/media/gooddz/xiaoliu/code/LSTR-main/晴天3检测结果/'
IMG_DIR_4 = '/media/gooddz/xiaoliu/code/LSTR-main/晴天3检测结果/'
out_put_video_name = 'output.avi' #输出视频的名称,
IMG_DIR = [IMG_DIR_1, IMG_DIR_2, IMG_DIR_3, IMG_DIR_4]
file_type = '.jpg' #图片后缀名为 .jpg,可以调整
fps = 60 #视频的FPS,可以适当调整
size=(1280,720) # 输出视频的分辨率,可以适当调整
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
for dir_num in range(0,len(IMG_DIR)):
IMG_DIR_temp = IMG_DIR[dir_num]
out_video_file_name = IMG_DIR_temp+out_put_video_name #设置视频的输出路径为 原文件夹下+前面设置的输出视频的名称
if os.path.exists(out_video_file_name): #判断视频是否已存在
os.remove(out_video_file_name)
videoWriter = cv2.VideoWriter(out_video_file_name,fourcc,fps,size)#最后一个是保存图片的尺寸
image_list = [f for f in os.listdir(IMG_DIR_temp) if f.endswith(file_type)]
image_list.sort(key= lambda x:int(x[:-4]))
for i in range(0, len(image_list)):
frame = cv2.imread(IMG_DIR_temp+image_list[i])
videoWriter.write(frame)
videoWriter.release()
2. 视频转成图片
import cv2
import argparse
import os
def parse_args():
"""
Parse input arguments
"""
parser = argparse.ArgumentParser(description='Process pic')
parser.add_argument('--input', help='video to process', dest='input', default=None, type=str)
parser.add_argument('--output', help='pic to store', dest='output', default=None, type=str)
#default为间隔多少帧截取一张图片
parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=1, type=int) #此处可更改提取帧的间隔
args = parser.parse_args(['--input','/media/gooddz/xiaoliu/lab数据集/test1.avi','--output','/media/gooddz/xiaoliu/lab数据集/1/']) #此处添加路径,input为输入视频的路径 ,output为输出存放图片的路径
return args
def process_video(i_video, o_video, num):
cap = cv2.VideoCapture(i_video)
num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT)
expand_name = '.jpg'
if not cap.isOpened():
print("Please check the path.")
cnt = 0
count = 0
while 1:
ret, frame = cap.read()
if not ret:
break
frame = cv2.resize(frame,(1280,720)) # 在尺寸可以修改图片尺寸
cnt += 1
if cnt % num == 0:
count += 1
print(os.path.join(o_video, str(count) + expand_name))
cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame)
if __name__ == '__main__':
args = parse_args()
if not os.path.exists(args.output):
os.makedirs(args.output)
print('Called with args:')
print(args)
process_video(args.input, args.output, args.skip_frame)
参考链接