我正在尝试NiftyNet,但在第一步中遇到了困难。
尝试运行quickstart命令


  python net_download.py density_vnet_abdominal_ct_model_zoo
  python net_segment.py推断-c〜/ niftynet / extensions / dense_vnet_abdominal_ct / config.ini


给我


  KeyError:“使用名称'FloorMod'注册两个渐变!(先前的注册在_find_and_load_unlocked:955中)”


有谁可以帮忙吗?我正在使用带有Nvidia GPU的Ubuntu 16.04。尝试过tensorflow:1.4.1-py3 docker镜像,Anaconda带有CPU版本的tensorflow
和带有tensorflow CPU版本的本地python,我得到了相同的错误。

我很确定这是我做错的事情,因为我在不同的环境中遇到了相同的错误,但是我不确定是什么...

谢谢!

最佳答案

我认为这是从TensorFlow 1.3(正式的NiftyNet依赖项)到1.4的变化。您可以切换回TF1.3,或在niftynet / layer / resampler.py中注释掉以下几行

@tf.RegisterGradient('FloorMod')
def _floormod_grad(op, grad):
    return [None, None]

关于python - 尝试运行NiftyNet快速启动命令时出错,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/48481327/

10-12 06:05