UnboundLocalError:の解決方法

Question

画像を読み込む関数を作成し，ディレクトリ内の画像を読み込んでfilename名と画像listを返すようにしました．
UnboundLocalError: local variable 'files' referenced before assignmentが出てしまい，原因と解決方法がわかりません．
ご教授をお願い致します．

  imglist=[]

  exclude_prefixes=('_','.')

  for root,dirs,files in os.walk(inputpath):
    dirs[:]=[dir for dir in dirs if 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]

      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)


#画像の読み込み
image_train,image_train_filenames=load_images('/content/drive/My Drive/reso/images/train',512,'Gray')
image_test,image_test_filenames=load_images('/content/drive/My Drive/reso/imagse/test',512,'Gray')

UnboundLocalError                         Traceback (most recent call last)
<ipython-input-96-1224aca6d0ba> in <module>()
      1 #画像の読み込み
      2 image_train,image_train_filenames=load_images('/content/drive/My Drive/reso/images/train',512,'Gray')
----> 3 image_test,image_test_filenames=load_images('/content/drive/My Drive/reso/imagse/test',512,'Gray')

<ipython-input-91-ad1e9c8b4772> in load_images(inputpath, imagesize, type_color)
     31   imgsdata=np.asarray(imglist,dtype=np.float32)
     32 
---> 33   return imgsdata,sorted(files)

UnboundLocalError: local variable 'files' referenced before assignment
コード
```

Answer 1

エラーの通り、return imgsdata,sorted(files)するときに、filesという変数が代入される前に参照されてるよ！！というものです。

filesはあくまで for root,dirs,files in os.walk(inputpath):のfor文を実行するときに定義されたこのfor文のループの中でのみ使用できる変数です。

そのため、ループを抜けたあとのreturn imgsdata,sorted(files)では変数の定義が見つからず、ローカル変数が代入されてへんよ！！ってなるわけです。

for文の前に別の変数をおいて、for文抜ける前に宣言した変数にfilesの内容を都度代入してあげるなどすればいいかと思います。

Comments

[tadapolice]

ありがとうございます！