im2col関数の実行におけるattributeErrorの改善方法を教えてください!

Question

import numpy as np

def im2col(images,flt_h,flt_w,out_h,out_w,stride,pad):
       n_bt,n_ch,img_h,img_w=images.shape
        
      
       img_pad=np.pad(images,[(0,0),(0,0),(pad,pad),(pad,pad)],"constant")
    
       cols=np.zeros((n_bt,n_ch,flt_h,flt_w,out_h,out_w))
        
       for h in range(flt_h):
             h_lim=h+stride*out_h
    
       for w in range(flt_w):
             w_lim=w+stride*out_w
             cols[:,:,h,w,:,:]=img_pad[:,:,h:h_lim:stride,w:w_lim:stride]
            
       cols=cols.transpose(1,2,3,0,4,5).reshape(
                          n_ch*flt_h*flt_w,n_bt*out_h*out_w)
       return cols


img=([[[[1,2,3,4],
            [5,6,7,8],
            [9,10,11,12],
            [13,14,15,16]]]])
cols=im2col(img,2,2,3,3,1,0)
print(cols)

発生している問題・エラーメッセージ

AttributeError                            Traceback (most recent call last)
<ipython-input-2-36a0f56e27d7> in <module>()
      3             [9,10,11,12],
      4             [13,14,15,16]]]])
----> 5 cols=im2col(img,2,2,3,3,1,0)
      6 print(cols)

<ipython-input-1-5e2d443ee39f> in im2col(images, flt_h, flt_w, out_h, out_w, stride, pad)
      2 
      3 def im2col(images,flt_h,flt_w,out_h,out_w,stride,pad):
----> 4        n_bt,n_ch,img_h,img_w=images.shape
      5 
      6 

AttributeError: 'list' object has no attribute 'shape'

Answer 1

'list'というオブジェクトは、'shape'という属性を持っていません。
'shape'という属性を持っているのは、'np.ndarray'です。numpyの配列です。

img=([[[...]]])の型はlistです。type(img)が'list'を返すはずです。

おそらく4次元配列を定義してそのサイズを'shape'で取得したいのだと思いますので、img=(...)を img=np.array(...)に変えてみてください。
そうすると、type(img)は、np.ndarrayを返すはずです。これで'shape'を使うことができます。

Comments

[tomatoto11]

解答有難うございます!無事解消する事が出来ました!

[Matsui_hero]

あと、蛇足かもしれませんが、for w in range(flt_w):のブロックは一つインデントを入れないと(for h in range(flt_h):の中に入れないと)正常に動かないと思います。

Answer 2

下記で解消しますか？
リスト型のimgをNumPy配列ndarrayに変換

cols=im2col(np.array(img),2,2,3,3,1,0)

Comments

[tomatoto11]

有難うございました!助かりました!

Answer 3

エラーメッセージから推測するとim2col関数の第一引数imagesはリスト型ではなくnumpy.array型でなければいけないようです。

Comments

[tomatoto11]

皆さん有難うございました!無事解消する事が出来ました!