opencv处理视频(img2vid.py)

【核心代码】

#!/usr/bin/env python
import cv2
from cv2 import VideoWriter,VideoWriter_fourcc,imread,resize
import os
import pdb

#vid_list=sorted(os.listdir('/Users/admin/Desktop/activitynet/cropped_images/'))
vid_list=['vid1207']
for vid in vid_list:
    #print(vid)
    img_root="/Users/admin/Desktop/video-captioning/msvd_dataset/images/%s"%vid
    im_names=sorted(os.listdir(img_root))
    if im_names != []:
        #print(im_names)
        #Edit each frame's appearing time!
        #fps=2
        img=cv2.imread('/Users/admin/Desktop/video-captioning/msvd_dataset/images/%s/%s'%(vid,im_names[0]))
        try:
            h,w,_=img.shape
        except:
            pdb.set_trace()
        videoWriter=cv2.VideoWriter("/Users/admin/Desktop/video-captioning/msvd_dataset/%s.avi"%vid,
        fourcc=VideoWriter_fourcc(*'MJPG'),fps=5,frameSize=(w,h),apiPreference=0)
        #fourcc=VideoWriter_fourcc(*'MJPG'),fps=5,frameSize=(224,224),apiPreference=0)
        for im_name in im_names:
            try:
                frame=cv2.imread('/Users/admin/Desktop/video-captioning/msvd_dataset/images/%s/%s'%(vid,im_name))
                frame=cv2.resize(frame,(w,h))
                videoWriter.write(frame)
            except:
                continue
            
        videoWriter.release()
    print('\r Finished %s -------------- %s/%s'%(vid,vid_list.index(vid) 1,len(vid_list)))