img2video.py
#! /usr/bin/env python
import cv2
import os
import glob
def imgs2video(imgs_dir): # imgs_dir是图片所在文件夹路径
fps = 24 # 频率
fourcc = cv2.VideoWriter_fourcc(*'MJPG') # 格式
# 下面是视频存储路径
file_path = imgs_dir + 'zjc.avi'
# avi格式亲测可用, mp4也可以但是报错如下:
'''
OpenCV: FFMPEG: tag 0x47504a4d/'MJPG' is not supported with codec id 7 and format 'mp4 / MP4 (MPEG-4 Part 14)'
OpenCV: FFMPEG: fallback to use tag 0x7634706d/'mp4v'
'''
video_writer = cv2.VideoWriter(file_path, fourcc, fps, (720, 540)) # 名字, 格式Four-Character Codes, 频率, 大小
# no glob, need number-index increasing
imgs = glob.glob(os.path.join(imgs_dir, '*.jpg')) #
# print(imgs) # 列表
'''
['/home/.../00000305.jpg', ..., '/home/.../00000306.jpg']
乱顺的 从00000070.jpg到00000450.jpg
'''
for i in range(len(imgs)):
i = i + 70 # 这里根据个人情况调整, 因为我的图片是从00000070.jpg 到 00000450.jpg 所以+70
imgname = os.path.join(imgs_dir, '{:0>8}.jpg'.format(i)) # >表示右对齐左边填充, :0表示填充0, 8表示8位数
# print(imgname)
'''
/home/hp/zjc/nk_PyCharm/PyCharm_project/nk_DeepSortYolo/deep_sort_yolov3-master/nk_src/bluesheep_2video/00000070.jpg
.
.
.
/home/hp/zjc/nk_PyCharm/PyCharm_project/nk_DeepSortYolo/deep_sort_yolov3-master/nk_src/bluesheep_2video/00000450.jpg
'''
frame = cv2.imread(imgname) # 读取图片-->frame
video_writer.write(frame) # 视频写入frame
video_writer.release() # 释放video_writer
if __name__ == '__main__':
imgs2video(
'/home/hp/zjc/nk_PyCharm/PyCharm_project/nk_DeepSortYolo/deep_sort_yolov3-master/nk_src/bluesheep_2video/')
video2img.py
#! /usr/bin/env python
import cv2
import os
import glob
def video2imgs(videoPath, savePicturePath): # imgs_dir是图片所在文件夹路径
cap = cv2.VideoCapture(videoPath) # 视频路径
numFrame = 0 # 用于写出图片的命名
numFrame = numFrame # 从1开始,这根据自己需要调整
while True:
if cap.grab(): # 如果成功获取图片
flag, frame = cap.retrieve()
if not flag:
print('no flag: flag, frame = cap.retrieve()')
continue
else:
# cv2.imshow('video', frame)
numFrame += 1 # 用于写出图片的命名
n = str(numFrame) # 字符化
n = n.zfill(8) # 这样就是八位数, 前面自动补充0
newPath = savePicturePath + n + ".jpg" # 输出图片的具体路径 = 输出图片的文件夹 + 图片名字.jpg
print(n)
cv2.imencode('.jpg', frame)[1].tofile(newPath) # 关于这个[1]暂时没有具体研究, 可以查看相关资料
else:
print('no cap.grab()')
break
if cv2.waitKey(10) == 27:
break
if __name__ == '__main__':
video2imgs(
'/home/hp/zjc/nk_PyCharm/PyCharm_project/nk_DeepSortYolo/deep_sort_yolov3-master/nk_src/15fpshorse.mp4',
'/home/hp/zjc/nk_PyCharm/PyCharm_project/nk_DeepSortYolo/deep_sort_yolov3-master/nk_src/horse_2img/')