[python] np.stack / 여러 영상을 한 이미지의 여러 채널로 합치기(의료영상)

딥러닝에서 여러 modality의 영상을 인풋으로 넣고 싶을 때, 여러 영상을 하나로 합쳐 인풋으로 넣는 방법이 있다.

 

만약 T1 강조 영상의 dimension이 (100, 100, 100)이고, T2 강조 영상의 dimension이 (100, 100, 100)이라면 이 두 영상을 인풋으로 넣고 싶을 때 (100, 100, 100, 2)의 dimension을 가진 한 영상으로 합쳐서 인풋으로 넣을 수 있다는 것이다.

 

img1 = nib.load(img1).get_fdata()
img2 = nib.load(img2).get_fdata()

img_concat = np.stack((img1, img2), axis=-1)

 

이런 방법으로 하면 두 영상을 인풋 하나로 넣을 수 있다.