python – Numpy和16位PGM

使用numpy在Python中读取16位PGM图像的有效且清晰的方法是什么？

我无法使用PIL加载16位PGM图像due to a PIL bug.我可以使用以下代码读取标题：

dt = np.dtype([('type', 'a2'),
               ('space_0', 'a1', ),
               ('x', 'a3', ),
               ('space_1', 'a1', ),
               ('y', 'a3', ),
               ('space_2', 'a1', ),
               ('maxval', 'a5')])
header = np.fromfile( 'img.pgm', dtype=dt )
print header

这打印出正确的数据:(‘P5′,”,’640′,”,’480′,”,’65535’)但我感觉这不是最好的方式.除此之外,我很难弄清楚如何通过16位读取x(y)(在这种情况下为640×480)以及大小(标题)的偏移量来读取以下数据.

编辑：图像添加

读取和显示图像的MATLAB代码是：

I = imread('foo.pgm'); 
imagesc(I);

看起来像这样：

解决方法:

import re
import numpy

def read_pgm(filename, byteorder='>'):
    """Return image data from a raw PGM file as numpy array.

    Format specification: http://netpbm.sourceforge.net/doc/pgm.html

    """
    with open(filename, 'rb') as f:
        buffer = f.read()
    try:
        header, width, height, maxval = re.search(
            b"(^P5\s(?:\s*#.*[\r\n])*"
            b"(\d+)\s(?:\s*#.*[\r\n])*"
            b"(\d+)\s(?:\s*#.*[\r\n])*"
            b"(\d+)\s(?:\s*#.*[\r\n]\s)*)", buffer).groups()
    except AttributeError:
        raise ValueError("Not a raw PGM file: '%s'" % filename)
    return numpy.frombuffer(buffer,
                            dtype='u1' if int(maxval) < 256 else byteorder+'u2',
                            count=int(width)*int(height),
                            offset=len(header)
                            ).reshape((int(height), int(width)))


if __name__ == "__main__":
    from matplotlib import pyplot
    image = read_pgm("foo.pgm", byteorder='<')
    pyplot.imshow(image, pyplot.cm.gray)
    pyplot.show()

标签：python,numpy,16-bit,pgm
来源： https://codeday.me/bug/20190926/1818244.html