from nibabel.viewers import OrthoSlicer3D
from nibabel import nifti1
import nibabel as nib
from matplotlib import pylab as plt
import matplotlib
# 文件名,nii或nii.gz
v_filename = '/home/luoshiyong/pycode/pythonProject/venv/lits/volume-65.nii'
s_filename = '/home/luoshiyong/pycode/pythonProject/venv/lits/segmentation-65.nii'
v_img = nib.load(v_filename)
s_img = nib.load(s_filename)
#print info of img
print(v_img)
#dimension
#print(v_img.dataobj.shape)
v_width,v_height,v_queue = v_img.dataobj.shape
s_width,s_height,s_queue = s_img.dataobj.shape
#print(s_img)
#print(s_img.dataobj.shape)
#
OrthoSlicer3D(v_img.dataobj).show()
OrthoSlicer3D(s_img.dataobj).show()
num = 1
for i in range(0,v_queue,30):
v_img_arr = v_img.dataobj[:,:,i]
plt.subplot(5,4,num)
plt.imshow(v_img_arr, cmap='gray')
num+=1
plt.show()