Spaces:
Configuration error
Configuration error
File size: 1,238 Bytes
d01f62c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | import sys
import os
from os import path
from shutil import copy2
input_path = sys.argv[1]
output_path = sys.argv[2]
multiplier = int(sys.argv[3])
image_path = path.join(input_path, 'JPEGImages')
gt_path = path.join(input_path, 'Annotations')
videos = sorted(os.listdir(image_path))
for vid in videos:
os.makedirs(path.join(output_path, 'JPEGImages', vid), exist_ok=True)
os.makedirs(path.join(output_path, 'Annotations', vid), exist_ok=True)
frames = sorted(os.listdir(path.join(image_path, vid)))
num_frames = len(frames)
counter = 0
output_counter = 0
direction = 1
for _ in range(multiplier):
for _ in range(num_frames):
copy2(path.join(image_path, vid, frames[counter]),
path.join(output_path, 'JPEGImages', vid, f'{output_counter:05d}.jpg'))
mask_path = path.join(gt_path, vid, frames[counter].replace('.jpg', '.png'))
if path.exists(mask_path):
copy2(mask_path,
path.join(output_path, 'Annotations', vid, f'{output_counter:05d}.png'))
counter += direction
output_counter += 1
if counter == 0 or counter == len(frames) - 1:
direction *= -1
|