Function apply

datasets/sampler.py:17–24

(data, method=None, frame_interval=None, seed=None, num_bucket=None)

# use pandarallel to accelerate bucket processing
# NOTE: pandarallel should only access local variables
def apply(data, method=None, frame_interval=None, seed=None, num_bucket=None):
    return method(
        data['num_frames'],
        data['height'],
        data['width'],
        frame_interval,
        seed + data['id'] * num_bucket,
    )


class StatefulDistributedSampler(DistributedSampler):