Ошибка при попытке запустить команду быстрого запуска NiftyNet

Question

Я пробовал NiftyNet и застрял на первом этапе. Попытка запустить команду быстрого запуска

python net_download.py density_vnet_abdominal_ct_model_zoo python net_segment.py вывод -c ~/niftynet/extensions/dens_vnet_abdominal_ct/config.ini

дает мне

KeyError: "Регистрация двух градиентов с именем 'FloorMod'!(Предыдущая регистрация была в _find_and_load_unlocked:955)"

Может ли кто-нибудь помочь? Я использую Ubuntu 16.04 с графическим процессором Nvidia. Пробный тензор потока: образ докера 1.4.1-py3, Anaconda с версией процессора tenorflow и нативный python с версией процессора тензор потока, и я получаю ту же ошибку.

Я почти уверен, что что-то сделал не так, потому что я получаю ту же ошибку из другой среды, но я не уверен, что...

Спасибо!

Answer 1

Я думаю, что это изменение от TensorFlow 1.3 (которая является официальной зависимостью NiftyNet) до 1.4. Вы можете вернуться к TF1.3 или закомментировать следующие строки в niftynet/layer/resampler.py

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

Answer 2

Пожалуйста, установите tenorflow с помощью этой команды pip install tensorflow После этого установите отличную сеть, используя команду ниже '' 'pip install niftynet' '' переустановите python '' 'pip install python' '', если проблема не исчезла, чем, пожалуйста, укажите вашу проблему в более подробности, пожалуйста, убедитесь, что ваша переменная среды установлена ​​перед выполнением команды со страницы niftynet.