のねのBlog

パソコンの問題や、ソフトウェアの開発で起きた問題など書いていきます。よろしくお願いします^^。

NameError: name 'StringIO' is not defined

Traceback (most recent call last):
  File "train.py", line 70, in <module>
    tf.app.run()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/app.py", line 125, in run
    _sys.exit(main(argv))
  File "train.py", line 66, in main
    flip_labels=args.flip_labels)
  File "/content/zi2zi/model/unet.py", line 539, in train
    for bid, batch in enumerate(train_batch_iter):
  File "/content/zi2zi/model/dataset.py", line 68, in batch_iter
    processed = [process(e[1]) for e in batch]
  File "/content/zi2zi/model/dataset.py", line 68, in <listcomp>
    processed = [process(e[1]) for e in batch]
  File "/content/zi2zi/model/dataset.py", line 40, in process
    img = bytes_to_file(img)
  File "/content/zi2zi/model/utils.py", line 26, in bytes_to_file
    return StringIO(bytes_img)
NameError: name 'StringIO' is not defined

kusanohitoshi.blogspot.com