Medpy AttributeError: 'Tensor' 對象沒有屬性 '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: 'Tensor' 對象沒有屬性 'astype'

Answer 1

astype是一個 Numpy 函數。 在這里，我猜result是tf.Tensor ，因此您需要使用tf.cast來更改其類型（請參閱https://www.tensorflow.org/api_docs/python/tf/cast ）。

Answer 2

tensor.numpy()可以將您的類型從 tensor 更改為numpy ，它將具有屬性astype