python - Medpy AttributeError：“张量”对象没有属性“astype”

Question

您好，我正在尝试使用 Medpy 和最新的 keras 和 tensorflow 库来计算平均表面距离（asd）。感谢您在这方面的任何帮助。这是我的代码和错误：

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/Segmentation/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：“张量”对象没有属性“astype”

标签： pythontensorflowkeras

Answer 1

astype是一个 Numpy 函数。这里我猜result是 a tf.Tensor，所以你需要改用tf.cast它来改变它的类型（见https://www.tensorflow.org/api_docs/python/tf/cast）。