应用蒙版Python枕头图像模块后使背景变黑

Question

我正在使用Tensorflow的对象检测API来检测图像中的对象并生成遮罩。他们提供的代码可以检测对象，生成蒙版并使图像的其余部分保持不变：

rgb = ImageColor.getrgb(color)
pil_image = Image.fromarray(image)

solid_color = np.expand_dims(
  np.ones_like(mask), axis=2) * np.reshape(list(rgb), [1, 1, 3])
pil_solid_color = 
Image.fromarray(np.uint8(solid_color)).convert('RGBA')
pil_mask = Image.fromarray(np.uint8(255.0*alpha*mask)).convert('L')
pil_image = Image.composite(pil_solid_color, pil_image, pil_mask)
np.copyto(image, np.array(pil_image.convert('RGB')))

如何修改此代码以完成其所有工作，而将背景设为黑色？