Function bucket_suffix

utils/dataset.py:59–70

(key)

def bucket_suffix(key):
    if len(key) == 2:
        # AR, frames
        return f'{key[0]:.{ROUND_DECIMAL_DIGITS}f}_{key[1]}'
    elif len(key) == 3:
        # width, height, frames
        return f'{key[0]}x{key[1]}x{key[2]}'
    elif len(key) == 4:
        # AR, width, height, frames
        return f'{key[0]:.{ROUND_DECIMAL_DIGITS}f}x{key[1]}x{key[2]}x{key[3]}'
    else:
        raise RuntimeError(f'Unexpected bucket: {key}')


def dedup_and_sort(values):