Question

我的共享字典对象的条目数不一致。它应该有500，但大多数测试最终都在450到465之间。我还尝试使用map和Process代替apply_async。

map略胜一筹，因为共享词典有大约480个条目而不是大约450个条目，但它仍然不一致，并且不是所有500个条目。

我也尝试使用Process，但这导致我的共享字典中的条目数量最少 - 大约420个。

以下是使用apply_async的完整代码：

import numpy as np
from PIL import Image
from os import listdir
from multiprocessing import Manager, Pool

def processImage(path, d):
  image = np.array(Image.open(source + "/" + path))

  # Copy lists from shared dictionary since updates don't work otherwise
  w = d["width"]
  h = d["height"]
  w.append(image.shape[0])
  h.append(image.shape[1])
  d["width"] = w
  d["height"] = h

if __name__ == "__main__":
  source = "./sample/images"
  p = Pool()
  m = Manager()
  d = m.dict()
  d["width"], d["height"] = [], []

  for path in listdir(source):
    p.apply_async(processImage, (path, d))
  p.close()
  p.join()

以下是使用map的完整代码：

def processImage(obj):
  image = np.array(Image.open(source + "/" + obj[1]))

  w = obj[0]["width"]
  h = obj[0]["height"]
  w.append(image.shape[0])
  h.append(image.shape[1])
  obj[0]["width"] = w
  obj[0]["height"] = h

if __name__ == "__main__":
  source = "./sample/images"
  p = Pool()
  m = Manager()
  d = m.dict()
  d["width"], d["height"] = [], []

  p.map(processImage, zip(itertools.repeat(d), listdir(source)))

以下是使用Process的完整代码：

def processImage(path, d):
  image = np.array(Image.open(source + "/" + path))

  w = d["width"]
  h = d["height"]
  w.append(image.shape[0])
  h.append(image.shape[1])
  d["width"] = w
  d["height"] = h

if __name__ == "__main__":
  source = "./sample/images"
  p = Pool()
  m = Manager()
  d = m.dict()
  d["width"], d["height"] = [], []

  jobs = []
  for img in listdir(source):
    p = Process(target=processImage, args=(img, d))
    p.start()
    jobs.append(p)

  for j in jobs:
    j.join()

Answer 1

这是竞争条件的典型例子。您需要某种同步原语来更新d。

考虑以下情况：执行processImage有两个线程（在您的情况下是子进程）。首先获得w和h，第二次获得w和h。首先向两者添加内容并将其放回d。第二个对自己的w和h做了一些事情，它不再考虑第一个线程所做的更改，并将其放回d。此时，第一个线程所做的更改将丢失。

要解决此问题，您需要保护使用d的代码部分：

from multiprocessing import Manager, Pool, Lock
...
lock = Lock()
...
def processImage(path, d):
    image = np.array(Image.open(source + "/" + path))

    lock.acquire()
    d["width"].append(image.shape[0])
    d["height"].append(image.shape[1])
    lock.release()

Python多处理池/进程具有不一致的结果

1 个答案: