RGB图像保存格式问题

原创于 2021-09-07 09:09:50 发布 · 1k 阅读

CC 4.0 BY-SA版权

文章标签：

本文探讨了在深度学习图像处理后，使用PIL的save方法和OpenCV的imwrite方法保存图片的差异。转换过程中，发现save保存的图片文件大小小于imwrite，可能由于fromarray函数的压缩。在保存前，save需要将array转为image，而imwrite则要求BGR通道顺序。这两种方法各有特点，适用于不同的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

完成深度学习的图像处理任务后，尝试了save和imwrite两种保存方法，遇到了一些问题。

1. img.save(filename)

from PIL import Image
adv_sample = Image.fromarray(np.uint8(adv_sample))
adv_sample.save(filename+'.jpg',quality=95)

处理完的adv_sample是array格式，而save需要保存image，因此需要先格式转换再保存。

2. cv2.imwrite(filename,img)

import cv2
adv_sample = adv_sample[:, :, ::-1]
cv2.imwrite('advSamples_images/images/' + dirname + '/' + filename + '.jpg',adv_sample,[cv2.IMWRITE_JPEG_QUALITY,100])

imwrite写入的是array格式，但是对于彩色图像来说，其输入的三通道顺序为"BGR"，因此需要先进行通道转换再保存。

以上两种方法进行对比，发现save保存的图片占用空间比imwrite小很多，有可能是fromarray函数对图片进行了压缩。