画像を読み込む関数を作成したい

Question

画像を読み込む関数（load_images()）を作成して画像を読み込むと，
結果がarray([], dtype=float32)になってしまいます．

load_imagesの中に問題がありますでしょうか？

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.startswich(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.array(testimage,dtype=np.float64)
          testimage=testimage[:,:,::-1]

        elif type_color=='Gray':
          testimage=cv2.imread(filename,cv2.IMREAD_GRAYSCALE)
          height,width=testimage.shape[:2]
          testimage=resize(testimage,(imagesize,imagesize),interpolation=cv2.INTER_AREA)
          testimage=np.array([testimage],dtype=np.float64)
          testimage=np.asarray(testimage,dtype=np.float64).reshape((imagesize,imagesize,1))

        imgslist.append(testimage)
    imgsdata=np.asarray(imglist,dtype=np.float32)

    return imgsdata,sorted(files)

image_train,image_train_filenames=load_images("drive/My Drive/denoise/image/train",256,'Gray')

image_test,image_test_filenames=load_images("drive/My Drive/denoise/image/test",256,'Gray')

image_train
⇒array([], dtype=float32)

load_images("drive/My Drive/denoise/image/train",256,'Gray')
⇒(array([], dtype=float32),
['train1.jpg', 'train2.jpg', 'train3.jpg', 'train4.jpg'])

Comments

[x98000]

で、load_images 関数はどこにあるのでしょうか

[tadapolice]

返信ありがとうございます．
質問の内容を編集しました．

[x98000]

で、結果のimage_train はどこで取得しているのでしょうか。

Answer 1

まず、コードレビューをしっかりやりましょう。エラーが出なければ正しいとは限りません。
また、想定した結果が得られなければ、各変数が想定通りかprint(変数)を適宜入れて確認していくのも有効です。
本ソースの原因はimgslistとimglistのタイプミスと思われます。再度確認してください。

Comments

[tadapolice]

返信ありがとうございます．
タイプミス修正し，
load_images("drive/My Drive/denoise/image/train",256,'Gray')で実行すると
⇒(array([], dtype=float32),
['train1.jpg', 'train2.jpg', 'train3.jpg', 'train4.jpg'])
になります．
やはり，画像の読み込みができていないでしょうか？

[x98000]

タイプミスが修正されていないので判断出来ません。また、ソースの修正は下に表示されるプレビュー画面を見て正しく修正してください。質問で指摘したimage_trainも無関係なら削除してください。