ADNI-python实现批量nii和dcm文件转换为png图像

需求:
转换后的PNG文件保存在与原始保存DCM和nii文件的文件夹格式相同的子文件夹中。并加入进度和时间预估功能。

任务步骤:
1.读取DCM和nii文件所在的目录树。
2.将每个DCM和nii文件转换为PNG并将其保存在与DCM和nii文件所在的子文件夹中的一个新文件夹中。
3.确保PNG文件名与其对应的DCM和nii文件名相同。

最终代码链接🔗notion记录:
修改文件夹所在地址就可以使用,但需要注意需要提前安装的库
python实现批量nii和dcm文件转换为png图像

DCM->PNG

import os
import numpy as np
import pydicom
from PIL import Image

# DCM文件路径
dcm_path = "/path/to/dcm/files/"
# PNG文件保存路径
png_path = "/path/to/png/files/"

# 获取所有DCM文件的路径
dcm_files = []
for dirpath, dirnames, filenames in os.walk(dcm_path):
    for filename in filenames:
        if filename.endswith(".dcm"):
            dcm_files.append(os.path.join(dirpath, filename))

# 遍历所有DCM文件并转换为PNG
for dcm_file in dcm_files:
    # 加载DCM文件
    ds = pydicom.dcmread(dcm_file)
    pixel_array = ds.pixel_array
    # 将像素数组转换为图像
    im = Image.fromarray(np.uint8(pixel_array))
    # 获取DCM文件所在的子文件夹名称
    sub_dirname = os.path.basename(os.path.dirname(dcm_file))
    # 创建PNG文件夹
    png_subdir = os.path.join(png_path, sub_dirname)
    os.makedirs(png_subdir, exist_ok=True)
    # 保存PNG文件
    png_filename = os.path.splitext(os.path.basename(dcm_file))[0] + ".png"
    png_file = os.path.join(png_subdir, png_filename)
    im.save(png_file)

我们首先遍历dcm_path目录树中的所有DCM文件,并将它们存储在一个列表中。然后,我们对每个DCM文件进行处理:我们读取像素数组并将其转换为PIL图像,然后根据DCM文件所在的子文件夹创建一个新的PNG文件夹,并将PNG文件保存到该文件夹中。

请注意,在上面的代码中,我们使用了os.walk()函数来遍历整个目录树,并使用os.path.dirname()和os.path.basename()函数从路径中提取子文件夹名称和文件名。我们还使用了os.makedirs()函数来创建PNG文件夹(如果尚不存在的话)。

因为图像数据太多,我添加了进度和时间预估功能

import os
import numpy as np
import pydicom
from PIL import Image
from tqdm import tqdm

# DCM文件路径
dcm_path = "/path/to/dcm/files/"
# PNG文件保存路径
png_path = "/path/to/png/files/"

# 获取所有DCM文件的路径
dcm_files = []
for dirpath, dirnames, filenames in os.walk(dcm_path):
    for filename in filenames:
        if filename.endswith(".dcm"):
            dcm_files.append(os.p
  • 5
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值