resize.py

'''
resize video data to 96x96
require ffmpeg
need to set [Walk, Run, ...] directories under raw_data/
    downloaded from http://www.wisdom.weizmann.ac.il/%7Evision/SpaceTimeActions.html
'''
import os
import glob

current_path = os.path.dirname(__file__)
resized_path = os.path.join(current_path, 'resized_data')
dirs = glob.glob(os.path.join(current_path, 'raw_data/*'))
files = [ glob.glob(dir+'/*') for dir in dirs ]
files = sum(files, []) # flatten

''' script for cropping '''
for i, file in enumerate(files):
    os.system("ffmpeg -i %s -pix_fmt yuv420p -vf crop=96:96:42:24 %s.mp4" %
             (file, os.path.join(resized_path, str(i))))

''' script for reducing size '''
# # resize to 96x76
# for i, file in enumerate(files):
#     os.system("ffmpeg -i %s -pix_fmt yuv420p -vf scale=96:-2 %s.mp4" %
#              (file, os.path.join(resized_path, str(i))))

# files = glob.glob(resized_path+'/*')
# for i, file in enumerate(files):
#     os.system("ffmpeg -y -i %s -pix_fmt yuv420p -vf pad=96:96:0:5 %s" %
#              (file, file))