エラーコード：too many indices for array: array is 2-dimensional, but 3 were indexed

Question

前提・実現したいこと

下記のエラーが発生しました。
インデックスの数は同じだと思うんですが、何が原因なのかわかりません。
解決策はありますでしょうか。
よろしくお願いします。

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

runfile('C:/Users/kanam/segmentation1/super resolution.py', wdir='C:/Users/kanam/segmentation1')
Traceback (most recent call last):

  File "C:\Users\kanam\segmentation1\super resolution.py", line 74, in <module>
    save_images("./train_low/", image_train_filenames, image_train)

  File "C:\Users\kanam\segmentation1\super resolution.py", line 62, in save_images
    testimage = testimage[:,:,::-1]#色チャンネルをRGBからBGRへ変換

IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed

該当のソースコード

import os
import cv2
import numpy as np

def load_images(inputpath, imagesize, type_color):
    imglist = []
    
    exclude_prefixes = ('__', '.')
    
    for root, dirs, files in os.walk(inputpath):
        dirs[:] = [dir for dir in dirs if not dir.startswith(exclude_prefixes)]
        files[:] = [file for file in files if not file.startswith(exclude_prefixes)]
        for fn in sorted(files):
            bn, ext = os.path.splitext(fn)
            if ext not in [".bmp", ".BMP", ".jpg", ".JPG", ".jpeg", ".JPEG", ".png", ".PNG"]:
                continue
                
            filename = os.path.join(root, fn)
            
            if type_color == 'Color':
                #カラー画像の場合
                testimage = cv2.imread(filename, cv2.IMREAD_COLOR)
                height, width = testimage.shape[:2] #サイズ変更
                testimage = cv2.resize(testimage, (imagesize, imagesize),
                                      interpolation = cv2.INTER_AREA)
                testimage = np.asarray(testimage, dtype=np.float64)
                testimage = testimage[:,:,::-1] #チャンネルをBGRからRGBに変更
                
            elif type_color == 'Gray':
                #グレースケール画像の場合
                testimage = cv2.imread(filename, cv2.IMREAD_GRAYSCALE)
                height, width = testimage.shape[:2] #サイズ変更
                testimage = cv2.resize(testimage, (imagesize, imagesize),
                                      interpolation = cv2.INTER_AREA)
                testimage = np.asarray([testimage], dtype=np.float64)
                testimage = np.asarray(testimage,
                            dtype=np.float64).reshape((imagesize, imagesize, 1))
                
            imglist.append(testimage)
    imgsdata = np.asarray(imglist, dtype=np.float32)
    
    return imgsdata, sorted(files) #画像リストとファイル名のリストを返す

def generate_lowresolution(imagelist, scale):
    imagelist_out = []
    
    for i in range(0, len(imagelist)):
        height, width = imagelist[i].shape[:2]
        image_temp = cv2.resize(imagelist[i], (round(height/scale), round(width/scale)),
                     interpolation = cv2.INTER_AREA)
        image_temp = cv2.resize(image_temp, (height, width),
                     interpolation = cv2.INTER_CUBIC)
        imagelist_out.append(image_temp)
    imgsdata = np.asarray(imagelist_out, dtype=np.float32)
    
    return imgsdata

def save_images(savepath, filenamelist, imagelist):
    for i, fn in enumerate(filenamelist):
        filename = os.path.join(savepath, fn)
        testimage = imagelist[i]
        testimage = testimage[:,:,::-1]#色チャンネルをRGBからBGRへ変換
        cv2.imwrite(filename, testimage)

#画像読み込み
image_train, image_train_filenames = load_images("./org_train_s/", 256, 'Gray')
image_test, image_test_filenames = load_images("./org_test/", 256, 'Gray')

#画像の低解像度化
image_train = generate_lowresolution(image_train, 3.0)
image_test = generate_lowresolution(image_test, 3.0)

#画像保存
save_images("./train_low/", image_train_filenames, image_train)
save_images("./test_low/", image_test_filenames, image_test)

IMAGE_SIZE = 256

Answer 1

image_train, image_train_filenames = load_images("./org_train_s/", 256, 'Gray')
image_test, image_test_filenames = load_images("./org_test/", 256, 'Gray')

で読み込んでいるから、カラーではないので２次元配列ですね。それの３次元目を逆向きにしようとしてエラーになっています。

Comments

[tthr]

ありがとうございます。
助かりました。