1

我正在使用混淆矩阵来衡量我的分类器的性能。这个例子对我来说很好用（它来自这里），但我一直都在TypeError: Invalid dimensions for image data 

from numpy import *
import matplotlib.pyplot as plt
from pylab import *

conf_arr = [[50.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], [3.0, 26.0, 0.0, 0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 10.0, 0.0, 0.0, 0.0, 0.0], [4.0, 1.0, 0.0, 5.0, 0.0, 0.0, 0.0], [3.0, 0.0, 1.0, 0.0, 6.0, 0.0, 0.0], [0.0, 0.0, 0.0, 0.0, 0.0, 47.0, 0.0], [2.0, 0.0, 0.0, 0.0, 0.0, 0.0, 8.0]]

norm_conf = []
for i in conf_arr:
        a = 0
        tmp_arr = []
        a = sum(i,0)
        for j in i:
                tmp_arr.append(float(j)/float(a))
        norm_conf.append(tmp_arr)

plt.clf()
fig = plt.figure()
ax = fig.add_subplot(111)
res = ax.imshow(array(norm_conf), cmap=cm.jet, interpolation='nearest')
cb = fig.colorbar(res)
savefig("confmat.png", format="png")

我是 python 和 matplotlib 的新手。有什么帮助吗？

Matplot 版本是 1.1.1。这是完整的回溯：

在 res =... 我得到 

TypeError      Traceback (most recent call last)
    C:\Python27\lib\site-packages\SimpleCV\Shell\Shell.pyc in <module>()
    ----> 1 res = ax.imshow(array(norm_conf), cmap=cm.jet, interpolation='nearest')

C:\Python27\lib\site-packages\matplotlib\axes.pyc in imshow(self, X, cmap, norm, aspect, interpolation, alpha, vmin, vmax, origin, extent, shape, filternorm, filter
rad, imlim, resample, url, **kwargs)
   6794                        filterrad=filterrad, resample=resample, **kwargs)
   6795
-> 6796         im.set_data(X)
   6797         im.set_alpha(alpha)
   6798         self._set_artist_props(im)

C:\Python27\lib\site-packages\matplotlib\image.pyc in set_data(self, A)
    409         if (self._A.ndim not in (2, 3) or
    410             (self._A.ndim == 3 and self._A.shape[-1] not in (3, 4))):
--> 411             raise TypeError("Invalid dimensions for image data")
    412
    413         self._imcache =None

TypeError: Invalid dimensions for image data

SimpleCV:105> cb = fig.colorbar(res)

对于 print norm_conf 我现在得到结果： [[1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], [1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], [1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],...]]。我纠正了缩进问题。但我的困惑 .png 非常扭曲。此外，我应该如何继续标记矩阵中的正方形？

4

1 回答 1

2

这对我来说很好（matplotlib 1.1.1rc）。最初我希望你确认你的 matplotlib 版本并发布整个回溯——“回溯”是指 TypeError 行之前的几行，它们显示了导致错误的原因——这仍然是一个好主意，但我想我明白了什么问题可能是。

norm_conf如果以某种方式没有被填充，这是你会得到的错误（即norm_conf = []）：

Traceback (most recent call last):
  File "mdim2.py", line 19, in <module>
    res = ax.imshow(array(norm_conf), cmap=cm.jet, interpolation='nearest')
  File "/usr/lib/pymodules/python2.7/matplotlib/axes.py", line 6796, in imshow
    im.set_data(X)
  File "/usr/lib/pymodules/python2.7/matplotlib/image.py", line 411, in set_data
    raise TypeError("Invalid dimensions for image data")
TypeError: Invalid dimensions for image data

您的代码看起来可能有一些缩进问题，这在使用混合制表符和空格时经常发生。因此，我建议 (1) 尝试python -tt yourprogramname.py查看是否存在空格错误，以及 (2) 确保您始终使用 4 空格制表符。

于 2012-09-19T13:03:20.273 回答