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