ฉันจะบันทึกภาพด้วย PIL ได้อย่างไร?

ฉันเพิ่งทำการประมวลผลรูปภาพโดยใช้ไลบรารีรูปภาพ Python (PIL) โดยใช้โพสต์ที่ฉันพบก่อนหน้านี้เพื่อทำการแปลงฟูเรียร์ของรูปภาพและฉันไม่สามารถให้ฟังก์ชันบันทึกทำงานได้ รหัสทั้งหมดใช้งานได้ดี แต่จะไม่บันทึกภาพที่ได้:

from PIL import Image
import numpy as np

i = Image.open("C:/Users/User/Desktop/mesh.bmp")
i = i.convert("L")
a = np.asarray(i)
b = np.abs(np.fft.rfft2(a))
j = Image.fromarray(b)
j.save("C:/Users/User/Desktop/mesh_trans",".bmp")

ข้อผิดพลาดที่ฉันได้รับมีดังต่อไปนี้:

save_handler = SAVE[string.upper(format)] # unknown format
    KeyError: '.BMP'

ฉันจะบันทึกภาพด้วย Pythons PIL ได้อย่างไร?

ข้อผิดพลาดเกี่ยวกับนามสกุลไฟล์ได้รับการจัดการแล้วคุณอาจใช้BMP(โดยไม่มีจุด) หรือส่งชื่อเอาต์พุตที่มีนามสกุลอยู่แล้ว ตอนนี้เพื่อจัดการกับข้อผิดพลาดที่คุณต้องแก้ไขข้อมูลของคุณอย่างถูกต้องในโดเมนความถี่เพื่อบันทึกเป็นรูปภาพจำนวนเต็มPILกำลังบอกคุณว่าไม่ยอมรับข้อมูลลอยเพื่อบันทึกเป็น BMP

นี่คือคำแนะนำ (พร้อมการปรับเปลี่ยนเล็กน้อยอื่น ๆ เช่นการใช้fftshiftและnumpy.arrayแทนnumpy.asarray) สำหรับการแปลงเพื่อการแสดงภาพที่เหมาะสม:

import sys
import numpy
from PIL import Image

img = Image.open(sys.argv[1]).convert('L')

im = numpy.array(img)
fft_mag = numpy.abs(numpy.fft.fftshift(numpy.fft.fft2(im)))

visual = numpy.log(fft_mag)
visual = (visual - visual.min()) / (visual.max() - visual.min())

result = Image.fromarray((visual * 255).astype(numpy.uint8))
result.save('out.bmp')

คุณควรจะสามารถให้ PIL รับประเภทไฟล์จากนามสกุลได้เช่นใช้:

j.save("C:/Users/User/Desktop/mesh_trans.bmp")

ลองลบ.ก่อนหน้า.bmp(ไม่ตรงBMPตามที่คาดไว้) ที่คุณสามารถดูจากความผิดพลาดที่save_handlerเป็นบนปลอกคุณให้ไว้แล้วมองสำหรับการแข่งขันในformat SAVEอย่างไรก็ตามคีย์ที่เกี่ยวข้องในออบเจ็กต์นั้นคือBMP(แทน.BMP)

ฉันไม่รู้อะไรมากPILนัก แต่จากการค้นหาอย่างรวดเร็วดูเหมือนว่าจะมีปัญหากับmodeภาพ การเปลี่ยนคำจำกัดความของjเป็น:

j = Image.fromarray(b, mode='RGB')

ดูเหมือนจะทำงานให้ฉัน (อย่างไรก็ตามโปรดทราบว่าฉันมีความรู้น้อยมากPILดังนั้นฉันขอแนะนำให้ใช้โซลูชันของ @ mmgp เพราะเขา / เขารู้อย่างชัดเจนว่าพวกเขากำลังทำอะไร :)) สำหรับประเภทต่างๆmodeฉันใช้หน้านี้ - หวังว่าหนึ่งในตัวเลือกจะเหมาะกับคุณ

ฉันรู้ว่านี่เก่า แต่ฉันพบว่า (ในขณะที่ใช้หมอน) เปิดไฟล์โดยใช้open(fp, 'w')แล้วบันทึกไฟล์จะทำงาน เช่น:

with open(fp, 'w') as f:
    result.save(f)

fp แน่นอนว่าเป็นเส้นทางของไฟล์