码迷,mamicode.com
首页 > 其他好文 > 详细

从一段视频中按帧提取图片

时间:2018-03-14 12:51:55      阅读:237      评论:0      收藏:0      [点我收藏+]

标签:print   序列   write   ima   opencv   alt   lam   jpg   ram   

这里实现的是从一段视频中每10帧读取第10帧图片,代码如下:

# -*- coding:utf-8 -*-
#视频中按序列提取帧,获得训练数据
import cv2
import os

video_src_path = "D:/mtcnn"
video_save_path = "D:/mtcnn/mydata"

videos = os.listdir(video_src_path)
videos = filter(lambda x : x.endswith("mp4") , videos)

for each_video in videos:
    print each_video

    each_video_name , _ = each_video.split(.)
    m = video_save_path + / + each_video_name

    if(os.path.exists(m) == False):
        os.mkdir(m)  #新建“D:/mtcnn/Mathbook_pic/Mathbook/”文件夹
    each_video_save_full_path = os.path.join(video_save_path, each_video_name) + /

    each_video_full_path = video_src_path + / + each_video
    print each_video_full_path
    cap = cv2.VideoCapture(each_video_full_path)

    if False == cap.isOpened():
        print open video failed  
    else:
        print open video succeeded

    count = 0   #统计帧数
    frame_gap = 10  #间隔为10帧,每个10帧读取第10帧
    success = True
    i = 0
    #每隔10帧读取一帧
    while(success):
        success , frame = cap.read()
        i = i + 1
        if(i == frame_gap):  #每个10帧读取第10帧
            print success
            #print ‘Read a new frame:‘ , success
            params = []
            params.append(int(cv2.IMWRITE_JPEG_QUALITY))
            params.append(95)
            cv2.imwrite(each_video_save_full_path + each_video_name + _%d.jpg % count , frame , params)
            count = count + 1
            i = 0

    cap.release()
print OK!

 

注意的是,对于cv2库的VideoCapture函数,如果之前使用的是Pip 安装的opencv的python库,则没有安装FFmPeg,会导致无法正常打开视频,需要按下面步骤进行操作:

技术分享图片

参考链接:http://blog.csdn.net/yedasha/article/details/78128345

 

从一段视频中按帧提取图片

标签:print   序列   write   ima   opencv   alt   lam   jpg   ram   

原文地址:https://www.cnblogs.com/zf-blog/p/8566374.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!