1--完整项目地址
liujf69/Data-Processing/Kinectis_400
2--参考
Kinetics-400数据集简介及下载
3--简单处理代码
import os
import csv
import cv2def split_frame(raw_csv: str, root_video_path: str, root_output_path: str, debug: bool = True):csv_reader = csv.reader(open(raw_csv))for idx, row in enumerate(csv_reader): if (idx == 0):continue # ['label', 'youtube_id', 'time_start', 'time_end', 'split', 'is_cc'] label, youtube_id, time_start, time_end, split, is_cc = rowvideo_name = youtube_id + "_" + time_start.zfill(6) + "_" + time_end.zfill(6) + ".mp4"print("Process ", idx, " ", video_name)video_path = os.path.join(root_video_path, label, video_name)save_path = os.path.join(root_output_path, label, video_name.split(".")[0])if not os.path.exists(save_path):os.makedirs(save_path)cap = cv2.VideoCapture(video_path)frame_idx = 0ret = Truewhile ret:ret, rgb_img = cap.read() # read each frameif (not ret):breakcv2.imwrite(save_path + '/' + str(frame_idx) + '.jpg', rgb_img)frame_idx = frame_idx + 1if debug: # just process one videobreakif __name__ == "__main__":raw_csv = "./label/train_256.csv"root_video_path = "./raw-part/compress/train_256"root_output_path = "./output"split_frame(raw_csv = raw_csv, root_video_path = root_video_path, root_output_path = root_output_path, debug = True)print("All Done!")
