Python空数组中的多处理（HoG特性）返回

Question

尝试在一个大数据集上提取HoG特征-总共70K。 所以测试值为28x28x70000 我想用游泳池。尝试前10张图片

from multiprocessing import Pool

def get_hog(inputImg):
    fd,hi=hog(inputImg, orientations=4, pixels_per_cell=(4, 4),
                    cells_per_block=(1, 1), visualize=True, multichannel=False)
    return fd

with Pool(5) as p:
    V = p.map(get_hog,test_X[:,:,range(0,10)]))

我在V中的结果是空数组

可能传递数据的函数不正确？在

Answer 1

在python的多处理中， 应该始终尝试使用回调方法来存储进程返回的数据。 因此，对于您的示例，我们可以使用池并在图像列表上运行池（在您的例子中是test_X），并让回调处理池中每个进程返回的数据。 下面是示例代码：

import numpy as np
from multiprocessing import Pool
V=[]
test_X = np.transpose(test_x)
def get_hog(inputImg):
    fd,hi=hog(inputImg, orientations=4, pixels_per_cell=(4, 4),
                cells_per_block=(1, 1), visualize=True, multichannel=False)
    return fd

def poolCallback(returnDataFromPool):
    global V
    V.append(returnDataFromPool)

p = Pool(5)
for x in test_X:
    p.apply_async(get_hog ,args=(x,),callback=poolCallback)    
p.close()
p.join()