我正在尝试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/