mmclassification/mmcls/core/utils/misc.py

# Copyright (c) OpenMMLab. All rights reserved.
from functools import partial


def multi_apply(func, *args, **kwargs):
    pfunc = partial(func, **kwargs) if kwargs else func
    map_results = map(pfunc, *args)
    return tuple(map(list, zip(*map_results)))