1. 程式人生 > >numpy的ndarray數組如何reshape成固定大小

numpy的ndarray數組如何reshape成固定大小

proc 尺寸 n) volume shape ces com 直接 每一個

在做肺結節檢測的時候,遇到dicom文件reshape之後尺寸大小不一。因為大下不一,numpy.reshape又無法重塑成指定大小的。最後還是在一個大牛的代碼中找到了解決方法。

VL = np.load(r‘D:\pycharm\TEAMWORK\Preprocess_3D\imageOR.npy‘)
# 我的imageOR中,每一個文件除了3維的ndarray之外,還保存了標簽lab,所以下面寫成isometric_volume[0],所以如果你只有數組信息,直接將後面的[0]去掉即可
votal = []
for isometric_volume in VL:
print(isometric_volume[0],‘第一個‘)

mean = np.mean(isometric_volume[0]).astype(np.float32)
std = np.std(isometric_volume[0]).astype(np.float32)
volume_resized = scipy.ndimage.interpolation.zoom(isometric_volume[0],
np.divide(64, isometric_volume[0].shape),
mode=‘nearest‘)

volume_resized = (volume_resized.astype(np.float32) - mean) / (std + 1e-7)
z0, z1 = volume_resized.shape[0]//2, volume_resized.shape[0]
y0, y1 = 0, volume_resized.shape[1]//2
volume_resized = volume_resized[z0:z1, y0:y1, :]
votal.append([volume_resized , isometric_volume[1]])
print(volume_resized,‘volume_resized‘)

print(volume_resized.shape)
print(votal , ‘votal‘)


np.save(‘newimage.npy‘,votal)

numpy的ndarray數組如何reshape成固定大小