最近在搞视频二分类,
使用的是I3D,但是我只需要rgb,暂时不需要flow。
原先代码里写的有问题,视频可以用,但是我的数据集已经预处理过了,都是图片。
代码里图片部分出错,所以自己写了一个简单的,没必要那么花里胡哨。
后续研究他那flow是怎么生成的,再加进来。
代码还是比较简练的。
自己在图片预处理时候已经进行间隔取帧了,这部分也不需要了。
import os
from pathlib2 import Path
import cv2
import numpy as np
def rgb(image_dir,save_dir):
ls=[]