could not broadcast input array from shape (128,128,3) into shape (128,128)

在深度学习时候，我们从网上爬取了一些图像，但是这些图像存在灰度图和彩色图，那么在进行数据处理的时候呢，就会出现，不一致问题，灰色的图按照彩色的图就去处理就会有问题，所以我们需要在处理前，对灰色图进行过滤，讲灰色图过滤掉，或者分彩色和灰度不同方式去处理。下面代码是我的简单处理办法

def load_image(image_path,is_gray):
    if is_gray:
        image = Image.open(image_path).convert('L')
    else:
        image = Image.open(image_path).convert('RGB')