Spaces:
Running
Running
| # -*- coding: UTF-8 -*- | |
| '''================================================= | |
| @Project -> File localizer -> image_to_video | |
| @IDE PyCharm | |
| @Author fx221@cam.ac.uk | |
| @Date 07/09/2023 20:15 | |
| ==================================================''' | |
| import cv2 | |
| import os | |
| import os.path as osp | |
| import numpy as np | |
| from tqdm import tqdm | |
| import argparse | |
| from tools.common import resize_img | |
| parser = argparse.ArgumentParser(description='Image2Video', formatter_class=argparse.ArgumentDefaultsHelpFormatter) | |
| parser.add_argument('--image_dir', type=str, required=True) | |
| parser.add_argument('--video_path', type=str, required=True) | |
| parser.add_argument('--height', type=int, default=-1) | |
| parser.add_argument('--fps', type=int, default=30) | |
| def imgs2video(img_dir, video_path, fps=30, height=1024): | |
| img_fns = os.listdir(img_dir) | |
| # print(img_fns) | |
| img_fns = [v for v in img_fns if v.split('.')[-1] in ['jpg', 'png']] | |
| img_fns = sorted(img_fns) | |
| # print(img_fns) | |
| # 输出视频路径 | |
| # fps = 1 | |
| img = cv2.imread(osp.join(img_dir, img_fns[0])) | |
| if height == -1: | |
| height = img.shape[1] | |
| new_img = resize_img(img=img, nh=height) | |
| img_size = (new_img.shape[1], height) | |
| # fourcc = cv2.cv.CV_FOURCC('M','J','P','G')#opencv2.4 | |
| # fourcc = cv2.VideoWriter_fourcc('I','4','2','0') | |
| fourcc = cv2.VideoWriter_fourcc(*'MP4V') # 设置输出视频为mp4格式 | |
| # fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V') # 设置输出视频为mp4格式 | |
| videoWriter = cv2.VideoWriter(video_path, fourcc, fps, img_size) | |
| for i in tqdm(range(3700, len(img_fns)), total=len(img_fns)): | |
| # fn = img_fns[i].split('-') | |
| im_name = os.path.join(img_dir, img_fns[i]) | |
| print(im_name) | |
| frame = cv2.imread(im_name, 1) | |
| frame = np.flip(frame, 0) | |
| frame = cv2.resize(frame, dsize=img_size) | |
| # print(frame.shape) | |
| # exit(0) | |
| cv2.imshow("frame", frame) | |
| cv2.waitKey(1) | |
| videoWriter.write(frame) | |
| videoWriter.release() | |
| if __name__ == '__main__': | |
| args = parser.parse_args() | |
| imgs2video(img_dir=args.image_dir, video_path=args.video_path, fps=args.fps, height=args.height) | |