ImportError: невозможно импортировать имя 'Iterator'

Question

Я хочу запустить код сегментации (по этой ссылке в github: https://github.com/taigw/brats17) в Google Colab, и я столкнулся с этой ошибкой:

/usr/local/lib/python3.6/dist-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
Traceback (most recent call last):
  File "test.py", line 18, in <module>
    from tensorflow.contrib.data import Iterator
ImportError: cannot import name 'Iterator'

Answer 1

Попробуйте это:

import tensorflow as tf
Iterator = tf.data.Iterator

Answer 2

Я подозреваю, что это конфликт версий для тензорного потока. Пакет data был обновлен с tf.contrib.data до базового пакета tf.data. Скорее всего, вы используете более новую версию tenorsflow, для которой этот код не был разработан. Вы можете установить более старую версию tenorflow с pip install tensorflow==1.5 (например, с установкой версии 1.5).