|
''' |
|
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) |
|
|
|
|
