#!/usr/bin/env python
import cv2
from cv2 import VideoWriter,VideoWriter_fourcc,imread,resize
import os
img_root="/home/zhang/20201024/yolo"
#Edit each frame's appearing time!
fps=5
#fourcc=VideoWriter_fourcc(*"MJPG")
#videoWriter=cv2.VideoWriter("./Vid.avi",fourcc,fps,(720,720))
videoWriter = cv2.VideoWriter('yolo.avi', cv2.VideoWriter_fourcc('M','P','E','G'), fps, (720,720))
im_names=os.listdir(img_root)
#for im_name in range(len(im_names)):
for filename in os.listdir(img_root):
    #frame=cv2.imread(img_root+str(im_name)+'.jpg')
    imgname="/home/zhang/20201024/yolo/"+filename
    frame=cv2.imread(imgname)
    frame=cv2.resize(frame,(720,720))
    print imgname
    #cv2.imshow("frame", frame)
    videoWriter.write(frame)
    cv2.waitKey(1)
videoWriter.release()