V-Net的Tensorflow实现
这是用于3D医学成像分割的架构的Tensorflow实现。 该代码仅实现Tensorflow图,必须在训练程序中使用它。
网络的视觉表示
这是此代码实现的网络示例。
用法示例
from VNet import VNet
input_channels = 6
num_classes = 1
tf_input = tf.placeholder(dtype=tf.float32, shape=(10, 190, 190, 20, input_channels))
model = VNet(num_classes=num_classes, keep_prob=.7)
logits = model.network_fn(tf_input, is_training=True)
logits将具有[10, 190, 190, 20, 1] logits形状[10,
1