参加2018之江杯全球人工智能大赛 ：视频识别&问答（二）

先将视频切分成一帧帧图片，再做后续处理，代码如下：

import cv2
import os

def capturevideo(file,filedir):
    os.mkdir("image/" + filedir)
    cap = cv2.VideoCapture(file)
    ret,im = cap.read()
    count = 0
    while ret:
        ret,im = cap.read()
        if ret:
            cv2.imwrite("image/"+filedir+"/"+str(count)+".jpg",im)
            count = count+1
    cap.release()

rootdir = r"D:\ai\AIE04\VQADatasetA_20180815\train"
list = os.listdir(rootdir)
for i in range(len(list)):
    path = os.path.join(rootdir,list[i])
    filedir = list[i].replace(".mp4","")
    print(i,list[i])
    capturevideo(path,filedir)
print('finish')

后边的思路是把问题分类，不同的问题提取不同的视频帧进行训练。如问人穿的衣服是什么颜色，则先判断帧中是否有人，将包含人的帧对问题进行训练。

提取了所有帧，这个量就比较大，而且很多帧之间的差别非常小，是否需要再抽样需要思考。

如问题包含颜色，则不能取灰度，否则可以将帧转为灰度图。