Я получаю IOError: bad message length
при передаче больших аргументов функции map
. Как я могу избежать этого?
Ошибка возникает, когда я устанавливаю N=1500
или больше.
Код:
import numpy as np
import multiprocessing
def func(args):
i=args[0]
images=args[1]
print i
return 0
N=1500 #N=1000 works fine
images=[]
for i in np.arange(N):
images.append(np.random.random_integers(1,100,size=(500,500)))
iter_args=[]
for i in range(0,1):
iter_args.append([i,images])
pool=multiprocessing.Pool()
print pool
pool.map(func,iter_args)
В документах multiprocessing
есть функция recv_bytes
, которая вызывает IOError. Может быть, из-за этого? (https://python.readthedocs.org/en/v2.7.2/library/multiprocessing.html)
ИЗМЕНИТЬ
Если я использую images
как массив numpy вместо списка, я получаю другую ошибку: SystemError: NULL result without error in PyObject_Call
.
Немного другой код:
import numpy as np
import multiprocessing
def func(args):
i=args[0]
images=args[1]
print i
return 0
N=1500 #N=1000 works fine
images=[]
for i in np.arange(N):
images.append(np.random.random_integers(1,100,size=(500,500)))
images=np.array(images) #new
iter_args=[]
for i in range(0,1):
iter_args.append([i,images])
pool=multiprocessing.Pool()
print pool
pool.map(func,iter_args)
EDIT2 Фактическая функция, которую я использую:
def func(args):
i=args[0]
images=args[1]
image=np.mean(images,axis=0)
np.savetxt("image%d.txt"%(i),image)
return 0
Кроме того, iter_args
не содержат один и тот же набор изображений:
iter_args=[]
for i in range(0,1):
rand_ind=np.random.random_integers(0,N-1,N)
iter_args.append([i,images[rand_ind]])