opencv视频抽帧

很奇怪的一个问题,用opencv读入手机拍的竖向(正向)视频抽帧保存,结果保存的图片都是横着的,在windows下直接把图片向右旋转保存,但是用标注软件labelImg把“已经旋转正”的图片打开时显示图片还是横着的,所以就想从根本上把抽帧图片顺时针旋转90度变成竖向再保存,代码如下。

# -*- coding:utf8 -*-
import cv2
import os
import shutil

def get_frame_from_video(video_name, interval):
    """
    Args:
        video_name:输入视频名字
        interval: 保存图片的帧率间隔
    Returns:
    """

    # 保存图片的路径
    save_path = './60682570/'

    # is_exists = os.path.exists(save_path)
    # if not is_exists:
    #     os.makedirs(save_path)
    #     print('path of %s is build' % save_path)
    # else:
    #     shutil.rmtree(save_path)
    #     os.makedirs(save_path)
    #     print('path of %s already exist and rebuild' % save_path)

    # 开始读视频
    video_capture = cv2.VideoCapture(video_name)
    i = 0
    j = 0

    while True:
        success, frame = video_capture.read()
        trans_img = cv2.transpose(frame) # 图片顺时针旋转90度
        new_img = cv2.flip(trans_img, 1)
        i += 1
        if i % interval == 0:
            # 保存图片
            j += 1
            save_name = save_path + str(j) + '_' + str(i) + '.jpg'
            cv2.imwrite(save_name, new_img)
            print('image of %s is saved' % save_name)
        if not success:
            print('video is all read')
            break

if __name__ == '__main__':
    # 视频文件名字
    video_name = './viedo/60682570.3gp'
    interval = 20
    get_frame_from_video(video_name, interval)

opencv其他可能关于图片方向显示不正常的原因:
链接: https://www.cnblogs.com/csonezp/p/5564809.html
链接: https://blog.csdn.net/xuan_xuan_/article/details/100295306

你可能感兴趣的:(opencv,计算机视觉,python)