Image processing using multiprocessing in Python

Question

I have an image stack. I am trying to use multiprocessing to do some process on each image in stack to get new image, and then bitwise_or with the old image.

Since multiprocessing won't keep the original order, I pass image index as args.

Later I bitwise_or the new image and old image by a for loop.

Is there a way to avoid for-loop for bitwise_or to have a better performance? Thanks

import cv2 as cv
import numpy as np
from multiprocessing import Pool

def process_img(idx, img):
    new_img = do_something(img)
    return idx, new_img

def main()
    # ...

    output_stack = np.copy(img_stack)

    with Pool() as pool:
        args = zip(
            range(len(img_stack)),
            img_stack)

        new_imgs_with_idx = pool.starmap(process_img, args)

        for i, new_img in new_imgs_with_idx:  # Any better ways?
            output_stack[i] = cv.bitwise_or(output_stack[i], new_img)

    # ...

    return output_stack