自作関数を作ったらエラーになった。

Question

    # カラー画像の時Gだけ抜き取りたい
    if len(img.shape) == 3:
        h, w, ch = img.shape
        
        # RGBのGだけ抜き取りたい
        img_g = img[:,:,1]

imgは画像のRGB情報を持つnumpy行列です。
これを関数化しようとして、

def extract(x, y):
    # カラー画像の時Gだけ抜き取りたい
    if len(x.shape) == 3:
        h, w, ch = x.shape
        
        # RGBのGだけ抜き取りたい
        img_g = x[:,:,y]

img_g = extract(img, 1)
print(123456)
print(img_g)

こうしたら、以下の出力とエラーが出ました、どうすれば良いでしょう。
123456
None

AttributeError: 'NoneType' object has no attribute 'T'

つまり、Noneになってるみたいです、なぜでしょう。

Comments

[y_waiwai]

---> 43 for (img, imgt) in zip(img_g, img_g.T):

って提示のコードにはありませんよ

Answer 1

つまり、Noneになってるみたいです、なぜでしょう。

「extract(x,y)」には戻り値がないため、実行結果は「値None, タイプNoneType」となります。
なので、「img_g = extract(img, 1)」で代入され、printを行うと「None」となります。

Comments

[退会済みユーザー]

# 元画像のshape (PILではchはRGB)
# print(img.shape)

def extract(x, y):
# カラー画像の時Gだけ抜き取りたい
if len(x.shape) == 3:
h, w, ch = x.shape

# RGBのGだけ抜き取りたい
return x[:,:,y]

v_max, v_min = 300, 200

#g
img_g = extract(img, 1)
#b
img_b = extract(img, 2)
#r
img_r = extract(img, 0)

#g
eximg_g = extract(eximg, 1)
#b
eximg_b = extract(eximg, 2)
#r
eximg_r = extract(eximg, 0)

imgrows, lenrows, imgcolumns, lencolumns = [], [], [], []

def diff(x, y):
for (img, imgt) in zip(x, x.T):
rows = img[(v_min<img)&(v_max>img)]
columns = imgt[(v_min<imgt)&(v_max>imgt)]
imgrows.append(rows)
lenrows.append(len(rows))
imgcolumns.append(columns)
lencolumns.append(len(columns))
return lenrows,lencolumns

test_data_list = diff(img_g, 1) + diff(img_b, 2) + diff(img_r, 0)

training_data_list = diff(eximg_g, 1) + diff(eximg_b, 2) + diff(eximg_r, 0)

print(test_data_list)

print(training_data_list)

できたっぽいです。