データ型をuint8にするとNameError: name 'PIL' is not definedになってしまいます

Question

前提・実現したいこと

DCGAN-tensorflow-masterを使って自前の画像を学習させ、新しく画像生成したいです。
一つ前のエラーではデータ型が間違っていたようなので修正するためuint8にしてみたのですが、今度は別のエラーが発生してしまいました。

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

 File "main.py", line 120, in main
    dcgan.train(FLAGS)
  File "/Users/shimauchi_namiki/DCGAN-tensorflow-master/model.py", line 195, in train
    grayscale=self.grayscale) for sample_file in sample_files]
  File "/Users/shimauchi_namiki/DCGAN-tensorflow-master/model.py", line 195, in <listcomp>
    grayscale=self.grayscale) for sample_file in sample_files]
  File "/Users/shimauchi_namiki/DCGAN-tensorflow-master/utils.py", line 45, in get_image
    resize_height, resize_width, crop)
  File "/Users/shimauchi_namiki/DCGAN-tensorflow-master/utils.py", line 103, in transform
    resize_height, resize_width)
  File "/Users/shimauchi_namiki/DCGAN-tensorflow-master/utils.py", line 96, in center_crop
    return np.array(im.resize([resize_h, resize_w]), PIL.Image.BILINEAR)
NameError: name 'PIL' is not defined

該当のソースコード

def center_crop(x, crop_h, crop_w,
                resize_h=64, resize_w=64):
  if crop_w is None:
    crop_w = crop_h
  h, w = x.shape[:2]
  j = int(round((h - crop_h)/2.))
  i = int(round((w - crop_w)/2.))
  im = Image.fromarray((x[j:j+crop_h, i:i+crop_w] * 255).astype(np.uint8))　＃ここにuint8を追加しました
  return np.array(im.resize([resize_h, resize_w]), PIL.Image.BILINEAR)

def transform(image, input_height, input_width,
              resize_height=64, resize_width=64, crop=True):
  if crop:
    cropped_image = center_crop(
      image, input_height, input_width,
      resize_height, resize_width)
  else:
    im = Image.fromarray((image[j:j+crop_h, i:i+crop_w] * 255).astype(np.uint8)) #ここにuint8を追加しました
  return np.array(im.resize([resize_h, resize_w]), PIL.Image.BILINEAR)/127.5 - 1.

試したこと

DCGAN-tensorflow-masterwをGithubからダウンロードしました。
その中のutils.pyのファイルに記述されているコードです。

im = Image.fromarray(x[j:j+crop_h, i:i+crop_w])と
im = Image.fromarray((image[j:j+crop_h, i:i+crop_w])の２箇所にuint8を追加しました。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

DCGAN-tensorflow/utils.py を見ると、from PIL import Image とモジュールを import しているのですから、PIL.Image.BILINEAR とは参照できません。以下の2箇所を修正すれば、少なくとも表題のエラーは解消すると思います。

- PIL.Image.BILINEAR
+ Image.BILINEAR

GitHub の個人が公開しているコードは十分デバッグされているわけではないので、コード自体に問題があることは当然ありえます。

Comments

[namiki_s]

ありがとうございます。試してみます。