将转换并行应用于图像列表

时间:2019-04-18 12:58:33

标签: python python-multiprocessing

我正在尝试对列表中的每个图像并行应用相同的变换(HOG /模糊)。图像是使用cv2.imread导入的numpy ndarray,并保存到列表中。

当前,当我将图像列表传递给函数时,似乎传递了数组中的单个条目,即图像中的一个像素而不是整个图像。您能建议如何将整个阵列/映像传递给池中的每个工作线程吗?

谢谢!

# imports
from functools import partial
from skimage.feature import hog
from multiprocessing import Pool

def findhog(image):
    return hog(image, visualise=True)[1] # hog returns two things, only return the visualisation

with Pool(processes=4) as pool:
    hogs = pool.map(findhog, listofimages)

0 个答案:

没有答案