京东猪脸识别比赛数据预处理：用Python将视频每一帧提取存储为图片

echo_1994

于 2017-11-23 11:07:16 发布

阅读量5.1k

点赞数 2

分类专栏：人工智能比赛文章标签： python 数据预处理京东猪脸识别 Linux

本文链接：https://blog.csdn.net/echo_1994/article/details/78612257

版权

参加京东猪脸识别比赛，使用Python的cv2模块从30个视频中提取帧并按类别存储为图片。通过Linux命令删除空图片，完成数据预处理。

摘要由CSDN通过智能技术生成

最近参加京东的猪脸识别比赛，训练集是30个视频，需要将视频的每一帧提取出来存储为图片，存入对应的文件夹（分类标签）。

本例是直接调用了cv2 模块中的 VideoCapture。一次运行，大概10分钟，就能得到预处理后的分类图片了，具体代码如下。

视频每一帧提取存储为图片代码

#! encoding: UTF-8

import os

import cv2
import cv

videos_src_path='/sata_disk/E_office/zhouhongli/pig/train'
images_save_path='/sata_disk/E_office/zhouhongli/pig/frame'

videos = os.listdir(videos_src_path)
videos = filter(lambda x: x.endswith('mp4'), videos)