Làm cách nào để lưu ảnh bằng PIL?

Tôi vừa thực hiện một số xử lý hình ảnh bằng thư viện hình ảnh Python (PIL) bằng cách sử dụng một bài đăng mà tôi đã tìm thấy trước đó để thực hiện chuyển đổi fourier của hình ảnh và tôi không thể làm cho chức năng lưu hoạt động. Toàn bộ mã hoạt động tốt nhưng nó chỉ không lưu hình ảnh kết quả:

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")

Lỗi tôi nhận được là như sau:

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

Làm cách nào để lưu ảnh bằng Pythons PIL?

Lỗi liên quan đến phần mở rộng tệp đã được xử lý, bạn sử dụng BMP(không có dấu chấm) hoặc chuyển tên đầu ra với phần mở rộng đã có. Bây giờ để xử lý lỗi, bạn cần sửa đổi đúng cách dữ liệu của mình trong miền tần số để được lưu dưới dạng hình ảnh số nguyên,PIL nó cho bạn biết rằng nó không chấp nhận dữ liệu float để lưu dưới dạng BMP.

Đây là một gợi ý (với các sửa đổi nhỏ khác, như sử dụng fftshiftvà numpy.arraythay vì numpy.asarray) để thực hiện chuyển đổi để có hình ảnh phù hợp:

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')

Bạn có thể chỉ cần cho phép PIL lấy loại tệp từ phần mở rộng, tức là sử dụng:

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

Hãy thử loại bỏ .trước .bmp(nó không khớp BMPnhư mong đợi). Như bạn có thể nhìn thấy từ lỗi, save_handlerlà phía trên vỏ các formatbạn cung cấp và sau đó tìm kiếm một trận đấu trong SAVE. Tuy nhiên, khóa tương ứng trong đối tượng đó là BMP(thay vì.BMP ).

Tôi không biết nhiều về nó PIL, nhưng từ một số tìm kiếm nhanh xung quanh, có vẻ như đó là một vấn đề với modehình ảnh. Thay đổi định nghĩa của jthành:

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

Có vẻ hiệu quả với tôi (tuy nhiên lưu ý rằng tôi có rất ít kiến ​​thức về nó PIL, vì vậy tôi khuyên bạn nên sử dụng giải pháp của @ mmgp vì s / anh ấy biết rõ ràng họ đang làm gì :)). Đối với các loạimode , tôi đã sử dụng này trang - hy vọng một trong những lựa chọn sẽ làm việc cho bạn.

Tôi biết rằng điều này đã cũ, nhưng tôi nhận thấy rằng (trong khi sử dụng Pillow), mở tệp bằng cách sử dụng open(fp, 'w')và sau đó lưu tệp sẽ hoạt động. Ví dụ:

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

fp tất nhiên là đường dẫn tệp.