python - skimage.color rgba 转换为 rgb 的图像通过 matplotlib imsave 保存为 rgba

Question

我需要将 PNG H*W*4 rgba 图像转换为形状为 H*W*3 的 rgb 图像.

我可以做到这一点，但是当我保存它时，图像会再次保存为H*W*4这是代码片段:

for idx, image in enumerate(image_names):
    #matplotlib as mpi here I use plt for plotting and mpi for read
    rgba = mpi.imread(os.path.join(read_path,image))
    #convert to rgb using skimage.color as rtl,
    rgb = rtl.rgba2rgb(rgba)
    #change path of the image to be saved
    resized_path = os.path.join(os.path.sep,Ims,p[0],image)
    print(np.shape(rgb))#shape is printed (136,136,3)
    mpi.imsave(resized_path,rgb)

此后，当我再次阅读它时，它的形状再次是H*W*4 任何想法为什么？我猜 matplotlib imsave 有什么东西吗？

引用图片:

编辑更新后的代码如下:

for idx, image in enumerate(image_names):
    rgba = plt.imread(os.path.join(read_path,image))
    rgb = skimage.color.rgba2rgb(rgba)
    #original image name do not have ext and adding or removing 
    # does not effect
    resized_path = os.path.join(os.path.sep,basepath,image,".png")
    rgb = Image.fromarray((rgb*255).astype(np.uint8))
    rgb.save(resized_path)

出现以下错误:

    ValueError                                Traceback (most recent call last)
<ipython-input-12-648b9979b4e9> in <module>()
      6         print(np.shape(rgb))
      7         rgb = Image.fromarray((rgb*255).astype(np.uint8))
----> 8         rgb.save(resized_path)
      9     #mpi.imsave(resized_path,rgb)

/usr/local/lib/python2.7/dist-packages/PIL/Image.pyc in save(self, fp, format, **params)
   1809                 format = EXTENSION[ext]
   1810             except KeyError:
-> 1811                 raise ValueError('unknown file extension: {}'.format(ext))
   1812 
   1813         if format.upper() not in SAVE:

ValueError: unknown file extension:

解决方案下面解决的答案是正确的，上面唯一的问题是调整了路径大小，这里进行了更改:

resized_path = os.path.join(os.path.sep,Ims,p[0],image)
resized_path = (resized_path+".png")

Answer 1

Matplotlib pyplot.imsave 使用 alpha channel (即 RGBA)保存图像，与输入数组是否存在这样的 channel 无关。

但是，不会丢失信息，因为所有 alpha 值都是 1。因此你可以得到你的 RGB 图像

new_im = plt.imread("image.png")[:,:,:3]
# new_im.shape will be (<y>, <x>, 3)

如果您特别需要 RGB png 图像，则需要使用不同的方法来保存图像。

例如使用PIL

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

im = np.random.rand(120,120,3)

im2 = Image.fromarray((im*255).astype(np.uint8))
im2.save("image2.png")

new_im = plt.imread("image2.png")  
print (new_im.shape)
# prints  (120L, 120L, 3L)