วิธีการแปลงอาร์เรย์ NumPy เป็นภาพ PIL โดยใช้ matplotlib colormap

ฉันมีปัญหาง่ายๆ แต่ไม่สามารถหาวิธีแก้ปัญหาที่ดีได้

ฉันต้องการใช้อาร์เรย์ NumPy 2D ซึ่งแสดงถึงภาพสีเทาและแปลงเป็นภาพ RGB PIL ในขณะที่ใช้ colormaps ของ matplotlib

ฉันสามารถรับเอาต์พุต PNG ที่เหมาะสมได้โดยใช้pyplot.figure.figimageคำสั่ง:

dpi = 100.0
w, h = myarray.shape[1]/dpi, myarray.shape[0]/dpi
fig = plt.figure(figsize=(w,h), dpi=dpi)
fig.figimage(sub, cmap=cm.gist_earth)
plt.savefig('out.png')

แม้ว่าฉันจะปรับสิ่งนี้เพื่อให้ได้สิ่งที่ต้องการ (อาจใช้ StringIO เพื่อรับภาพ PIL) แต่ฉันสงสัยว่าไม่มีวิธีที่ง่ายกว่านี้ในการทำเช่นนั้นเนื่องจากดูเหมือนว่าจะเป็นปัญหาที่เป็นธรรมชาติมากในการสร้างภาพ สมมติว่าสิ่งนี้:

colored_PIL_image = magic_function(array, cmap)

ค่อนข้างยุ่งหนึ่งซับ แต่นี่คือ:

1. ครั้งแรกให้แน่ใจว่าอาร์เรย์ NumPy ของคุณเป็นปกติที่มีค่าสูงสุดที่myarray1.0
2. myarrayสมัครตารางสีโดยตรง
3. ปรับขนาดใหม่เป็น0-255ช่วง
4. แปลงเป็นจำนวนเต็มโดยใช้np.uint8().
5. ใช้Image.fromarray().

และเสร็จแล้ว:

from PIL import Image
from matplotlib import cm
im = Image.fromarray(np.uint8(cm.gist_earth(myarray)*255))

กับplt.savefig():

กับim.save():

• input = numpy_image
• np.unit8 -> แปลงเป็นจำนวนเต็ม
• แปลง ('RGB') -> แปลงเป็น RGB

• Image.fromarray -> ส่งคืนวัตถุรูปภาพ

  from PIL import Image
  import numpy as np
  
  PIL_image = Image.fromarray(np.uint8(numpy_image)).convert('RGB')
  
  PIL_image = Image.fromarray(numpy_image.astype('uint8'), 'RGB')

วิธีการที่อธิบายไว้ในคำตอบที่ยอมรับไม่ได้ผลสำหรับฉันแม้ว่าจะใช้การเปลี่ยนแปลงที่กล่าวถึงในความคิดเห็นแล้วก็ตาม แต่รหัสง่ายๆด้านล่างใช้งานได้:

import matplotlib.pyplot as plt
plt.imsave(filename, np_array, cmap='Greys')

np_array อาจเป็นอาร์เรย์ 2 มิติที่มีค่าตั้งแต่ 0..1 ลอยตัว o2 0..255 uint8 และในกรณีนี้จำเป็นต้องใช้ cmap สำหรับอาร์เรย์ 3 มิติ cmap จะถูกละเว้น