tools/data/k400/extract_frames.py [19:61]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        source_class_dir = os.path.join(source_dir, class_index)
        videos = os.listdir(source_class_dir)
        videos.sort()

        target_class_dir = os.path.join(target_dir, class_index)
        if not os.path.exists(target_class_dir):
            os.makedirs(target_class_dir)

        for each_video in videos:
            source_video_name = os.path.join(source_class_dir, each_video)
            video_prefix = each_video.split('.')[0]
            target_video_frames_folder = os.path.join(target_class_dir, video_prefix)
            if not os.path.exists(target_video_frames_folder):
                os.makedirs(target_video_frames_folder)
            target_frames = os.path.join(target_video_frames_folder, 'frame_%05d.jpg')

            try:
                # change videos to 30 fps and extract video frames
                subprocess.call('ffmpeg -nostats -loglevel 0 -i %s -filter:v fps=fps=30 -s 340x256 -q:v 2 %s' %
                                (source_video_name, target_frames), shell=True)

                # sanity check video frames
                video_frames = os.listdir(target_video_frames_folder)
                video_frames.sort()

                if len(video_frames) == 300:
                    # exactly 300 frames
                    continue
                elif len(video_frames) > 300:
                    # remove frames longer than 300
                    for i in range(300, len(video_frames)):
                        os.remove(os.path.join(target_video_frames_folder, video_frames[i]))
                else:
                    # duplicate videos with less than 300 frames
                    last_file = 'frame_%05d.jpg' % (len(video_frames) - 1)
                    last_file = os.path.join(target_video_frames_folder, last_file)
                    for i in range(len(video_frames), 300 + 1):
                        new_file = 'frame_%05d.jpg' % i
                        new_file = os.path.join(target_video_frames_folder, new_file)
                        shutil.copyfile(last_file, new_file)
            except:
                print('Video %s decode failed.' % (source_video_name))
                continue
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



tools/data/k400/extract_frames_parallel.py [10:52]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    source_class_dir = os.path.join(source_dir, class_index)
    videos = os.listdir(source_class_dir)
    videos.sort()

    target_class_dir = os.path.join(target_dir, class_index)
    if not os.path.exists(target_class_dir):
        os.makedirs(target_class_dir)

    for each_video in videos:
        source_video_name = os.path.join(source_class_dir, each_video)
        video_prefix = each_video.split('.')[0]
        target_video_frames_folder = os.path.join(target_class_dir, video_prefix)
        if not os.path.exists(target_video_frames_folder):
            os.makedirs(target_video_frames_folder)
        target_frames = os.path.join(target_video_frames_folder, 'frame_%05d.jpg')

        try:
            # change videos to 30 fps and extract video frames
            subprocess.call('ffmpeg -nostats -loglevel 0 -i %s -filter:v fps=fps=30 -s 340x256 -q:v 2 %s' %
                            (source_video_name, target_frames), shell=True)

            # sanity check video frames
            video_frames = os.listdir(target_video_frames_folder)
            video_frames.sort()

            if len(video_frames) == 300:
                # exactly 300 frames
                continue
            elif len(video_frames) > 300:
                # remove frames longer than 300
                for i in range(300, len(video_frames)):
                    os.remove(os.path.join(target_video_frames_folder, video_frames[i]))
            else:
                # duplicate videos with less than 300 frames
                last_file = 'frame_%05d.jpg' % (len(video_frames) - 1)
                last_file = os.path.join(target_video_frames_folder, last_file)
                for i in range(len(video_frames), 300 + 1):
                    new_file = 'frame_%05d.jpg' % i
                    new_file = os.path.join(target_video_frames_folder, new_file)
                    shutil.copyfile(last_file, new_file)
        except:
            print('Video %s decode failed.' % (source_video_name))
            continue
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -