Python jupyter笔记本中3d分割的绘图预测

Python jupyter笔记本中3d分割的绘图预测,python,matplotlib,deep-learning,3d,torch,Python,Matplotlib,Deep Learning,3d,Torch,开发了一种用于三维CT图像语义分割的神经网络。在每张图像中分割出7个不同的类别(即肾脏、肿瘤、肝脏…) 神经网络的输入是形状(1,1,132,132,116)-(批量大小,通道(灰度图像),宽度,高度,长度)的火炬张量 神经网络的输出是(7132132116)-颜色通道是第一维,然后是宽度、高度、长度 我正在寻找一种方法,如何在两个切片(但matplotlib至少应使用适当的颜色进行分割)或最好在3d体积中绘制,并将颜色设置为通道 可以绘制3d图像的切片,但是只能从nil.gz格式绘制,我没有这

开发了一种用于三维CT图像语义分割的神经网络。在每张图像中分割出7个不同的类别(即肾脏、肿瘤、肝脏…)

神经网络的输入是形状(1,1,132,132,116)-(批量大小,通道(灰度图像),宽度,高度,长度)的火炬张量

神经网络的输出是(7132132116)-颜色通道是第一维,然后是宽度、高度、长度

我正在寻找一种方法,如何在两个切片(但matplotlib至少应使用适当的颜色进行分割)或最好在3d体积中绘制,并将颜色设置为通道

可以绘制3d图像的切片,但是只能从nil.gz格式绘制,我没有这种格式(可以理解,我不想将张量保存为nil.gz格式,然后加载它,因为这是一个不必要的步骤)

上面的代码适用于一个分段类和一个切片,但是,我希望至少在该特定切片上分段所有类

谢谢

plt.imshow(pred[1][10],origin='lower')
plt.show()