【NumPy】tiffファイルからカラー付きの画像を出す

tiffファイルから画像（.png）を保存するプログラムを作成しています。
【tiffファイル内容】
1500×3500で２次元配列となっており，0~6までの数字が
値として入っています。データ型はnp.float32です。

このファイルを画像化して，pngに画像を保存したいと考えており，
カラーはPairedで，最小値0，最大値6に色付けした画像を作成したいのです。

●コード●

import os
import sys
import numpy
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import glob
from PIL import Image
import PIL

path = "/media/student/data1"
path2 = "/media/student/data1/ctype/"

datalist = glob.glob(path2 + "*.tif")

for data in datalist:
    name = data[len(path2)+2:-4]
    if name :
        print name + "..." + "OK"
	    cloud = np.fromfile(data,dtype=np.float32)
	    plt.imshow(cloud, cmap = 'Paired', vmin = 0, vmax = 6, extent=(130,180,0,30), interpolation='none')
	    pil_img = Image.fromarray(np.float32(cloud))
        print "cloud...ok"
        outdata = path + "/test/" +"ct-" +name + ".png"
        pil_img.save(outdata)

●エラー●

student@sh2:/media/student/data1$ python c_gazou.py
20160924-1800...OK
/media/student/data1/ctype/c-20160924-1800.tif
Traceback (most recent call last):
  File "c_gazou.py", line 27, in <module>
    plt.imshow(cloud, cmap = 'Paired', vmin = 0, vmax = 6, extent=(130,180,0,30), interpolation='none')
  File "/usr/lib/python2.7/dist-packages/matplotlib/pyplot.py", line 3022, in imshow
    **kwargs)
  File "/usr/lib/python2.7/dist-packages/matplotlib/__init__.py", line 1814, in inner
    return func(ax, *args, **kwargs)
  File "/usr/lib/python2.7/dist-packages/matplotlib/axes/_axes.py", line 4947, in imshow
    im.set_data(X)
  File "/usr/lib/python2.7/dist-packages/matplotlib/image.py", line 453, in set_data
    raise TypeError("Invalid dimensions for image data")
TypeError: Invalid dimensions for image data

plt.imshow(cloud, cmap = 'Paired', vmin = 0, vmax = 6, extent=(130,180,0,30), interpolation='none')
ここでエラーを起こしているみたいです。
エラーを解消するにはどのようにすればいいでしょうか？
よろしくお願いします。

行動規範の内容に同意します

回答1件

ベストアンサー

np.fromfileでTIFFは直接読めないんじゃないですかね（わかりませんが）。一回PILで読み込んでから配列に変換してみてはどうでしょう？

難航するようであれば、一枚画像のサンプルを頂けると回答する側もデバッグできます。もしTIFFがteratailにそのままアップロードできなければ、アップローダか何かに上げていただければ。

追記

案の定np.fromfileは読めてませんでした（滅茶苦茶な次元数のnumpy配列になってた。逆によくエラーも吐かず読み込むものです）。

とりあえず、意図している（であろう）変換をするコードを書いてみたので、参考にしてください。

python
1import numpy as np
2import matplotlib.cm as cm
3from matplotlib.colors import Normalize
4from PIL import Image
5
6pil_img = Image.open("img.tif")  # PIL.Imageで読む。ファイル名は変えました
7img_array = np.asarray(pil_img)  # arrayに変換
8
9norm = Normalize(vmin=0, vmax=6)  # matplotlibのnormは使う。参照：https://matplotlib.org/users/colormapnorms.html
10img_converted = cm.Paired(norm(img_array))*255  # matplotlibのcm.Pairedは使う。値のスケールが0~1なので0~255に変換
11out_img = Image.fromarray(np.uint8(img_converted))  # ここでplt.imshow使うのもなんなのでそのままPIL.Imageで処理することにした
12out_img.save("out_img.png")  # 出力