Medpy AttributeError: у объекта 'Tensor' нет атрибута 'astype'

Question

Здравствуйте, я пытаюсь вычислить среднее расстояние от поверхности (asd), используя Medpy с последними библиотеками керас и тензорного потока. Любая помощь в этом отношении приветствуется. Вот мой код и ошибка:

from medpy import metric as M

asd=M.binary.asd(tf.cast(y_pred, np.float32),tf.cast(y_true, np.float32),voxelspacing=None,connectivity=1)

Ошибка:

Файл "/ home / rak038 / Сегментация / U-net / noGPU / noGPU / lib / python3. 7 / site-packages / medpy / metric / binary.py ", строка 1200, в __surface_distances

result = numpy.atleast_1d(result.astype(numpy.bool))

AttributeError: у объекта 'Tensor' нет атрибута 'astype'

Answer 1

astype является функцией Numpy. Здесь я думаю, result - это tf.Tensor, поэтому вам нужно использовать tf.cast, чтобы изменить его тип (см. https://www.tensorflow.org/api_docs/python/tf/cast).