使用keras.preprocessing.image load_img时，图像的背景变黑的背景

Question

我的代码：

from keras.preprocessing.image import load_img, img_to_array
from keras.preprocessing.image import smart_resize
import numpy as np

#load pretrained model
os.chdir('path/to/dir')
# model = load_model(model_path)
input_imagesX = natsorted(os.listdir())
for idx, v in enumerate(input_imagesX):
    
    img = load_img(v,color_mode='rgb')
    img = img_to_array(img)
    img = tensorflow.image.resize(img, (128,128))
    # img = np.expand_dims(img, axis=0)
    plt.imshow(img)
    plt.grid(None)
    plt.show()

在这里，PNG的透明背景变成黑色，填充会变形。尝试了几个骇客，似乎没有任何作用！ 有人可以帮忙吗？

原始图像

我获得的图像

Answer 1

有些图像也可能具有Alpha通道。请在下面找到如何加载alpha通道

img = load_img("/content/transparent.png",color_mode='rgba')
img = img_to_array(img)
img = tensorflow.image.resize(img, (128,128))
img=tensorflow.cast(img,tensorflow.int32) 
plt.imshow(img)
plt.grid(None)
plt.show()

输出：

让我们知道问题是否仍然存在。谢谢！