【Python】MRI数据集处理 频域数据转换到空域

1. 原数据集格式为.npy，维度[C,H,W,F]，其中C为切片数，H，W为长和宽，F=2 是复数的实部和虚部。
2. 原数据集是频域数据[C,H,W,F]，按需求转换到空域[C,H,W]，且数据范围归一化到0-255。

import numpy as np
import os 

data_path = "./data/"
save_path = "./save/"

data_names = os.listdir(data_path)
for data_name in data_names:
    data = np.load(data_path + data_name)
    C,H,W,F = data.shape
    
    new_data = np.zeros((C,H,W))
    for i in range(C):
        a = np.empty([H,W], dtype=complex)
        a.real = data[i,:,:,0]
        a.imag = data[i,:,:,1]
        
        b = np.fft.ifft2(a)
        b = np.abs(b)
        
        b_max = np.max(b)
        b_min = np.min(b)
        b = (b - b_min) / (b_max - b_min) *255
        new_data[i,...] = b
    
    np.save(save_path + str(data_name), new_data)
print('finish.')