NiftyNet中的InvalidArgumentError

Question

我尝试在自己的MRI数据中测试NiftyNet的脑部分割情况。从NiftyNet repository下载经过预先​​训练的网络dense_vnet_abdominal_ct_model_zoo之后，使用以下命令将其用于从我们的MRI数据中分割出大脑。

net_segment inference -c config.ini

在此配置文件config.ini中，spatial_window_size设置为（112、112、112），在密集Vnet中应被8整除。

但是，我收到与Tensorflow相关的错误“ InvalidArgumentError：Assign需要两个张量的形状都匹配”。追溯列在下面。

File "\niftynet\engine\application_driver.py", line 261, in    
run_application
    self._rand_init_or_restore_vars(session)
File "\niftynet\engine\application_driver.py", line 402, in
_rand_init_or_restore_vars
    self.saver.restore(sess, checkpoint)

File "\niftynet\engine\application_driver.py", line 255, in
run_application
    self.graph = self._create_graph(self.graph)
File "\niftynet\engine\application_driver.py", line 356, in
_create_graph
    save_relative_paths=True)

尽管此问题以前在Tensorflow和NiftyNet中分别以issue 1，issue 2和issue 3的形式解决，但这些无助于解决问题。您能帮我解决这个错误吗？