| ''' |
| This file is to help us make figure of CRF vs Preset in Video Compression |
| ''' |
| import os, sys, shutil |
|
|
| def compress(input_folder, codec, crf, preset): |
| video_store_name = "compressed.mp4" |
| store_dir = input_folder + "_crf" + str(crf) + "_" + preset |
| |
| if os.path.exists(video_store_name): |
| os.remove(video_store_name) |
| if os.path.exists(store_dir): |
| shutil.rmtree(store_dir) |
| os.makedirs(store_dir) |
|
|
|
|
| |
| os.system("ffmpeg -r 30 -f image2 -i " + input_folder + "/%d.png -vcodec " + codec + " -crf " + str(crf) + " -preset " + preset + " -pix_fmt yuv420p " + video_store_name) |
|
|
| |
| os.system("ffmpeg -i " + video_store_name + " " + store_dir + "/test_%06d.png") |
|
|
|
|
|
|
| if __name__ == "__main__": |
| input_folders = ["ReadySetGo", "Jockey"] |
| codec = "libx264" |
| crf_ranges = [25 + 5*i for i in range(6)] |
| preset_ranges = ["ultrafast", "veryfast", "fast", "medium", "slow", "veryslow", "placebo"] |
|
|
|
|
| for input_folder in input_folders: |
| for crf in crf_ranges: |
| for preset in preset_ranges: |
| print("We are handling {} with crf {} with preset {}".format(input_folder, crf, preset)) |
| compress(input_folder, codec, crf, preset) |
|
|
| |
