make_frameACB.py

"""
Put it into the corresponding datasets directory, e.g. `/datasets/motion_mag_data/train/train_vid_frames` for me.
Make the original frames into frameAs, frameBs, frameCs(same as frameBs here)
"""
import os
import sys


# Choose the dir you want
dirs = sorted([i for i in os.listdir('.') if i in 
sys.argv[1].split('+')
# and int(i.split('_')[-1].split('.')[0]) > 0
]
# , key=lambda x: int(x.split('_')[-1])
)[:]

image_format_name='png'

for d in dirs:
    print('ACB-Processing on', d)
    os.chdir(d)
    os.mkdir('frameA')
    os.mkdir('frameC')
    files = sorted([f for f in os.listdir('.') if os.path.splitext(f)[1] == '.{}'.format(image_format_name)], key=lambda x: int(x.split('.')[0]))
    os.system('cp ./*{} frameA && cp ./*{} frameC'.format(image_format_name, image_format_name))
    os.remove(os.path.join('frameA', files[-1]))
    os.remove(os.path.join('frameC', files[0]))
    for f in sorted(os.listdir('frameC'), key=lambda x: int(x.split('.')[0])): 
        f_new = os.path.join('frameC', '%06d' % (int(f.split('.')[0])-1) + '.{}'.format(image_format_name))
        f = os.path.join('frameC', f)
        os.rename(f, f_new)
    os.system('cp -r frameC frameB')
    os.system('rm ./*.{}'.format(image_format_name))
    os.chdir('..')