nnunet入门之二 (2D图像分割)
创始人
2024-06-02 06:02:51
0

目录

  • 安装环境
  • 数据处理
  • 预处理
  • 训练
  • 测试

MIC-DKFZ/nnUNet
选择Linux环境运行该项目,Windows环境需要更改较多的参数,暂不支持。

安装环境

可以查看另一篇

数据处理

2D数据转成伪3D数据,即z轴层数为1或3(灰度图为1,彩图为3)。

参考:nnUNet/nnunet/dataset_conversion/Task120_Massachusetts_RoadSegm.py

直接转换数据,将数据存放到nnUNet_raw_data_base/nnUNet_raw_data/Task120_MassRoadsSeg

  1. 文件夹目录

    └─Task120_MassRoadsSeg│  dataset.json│  ├─imagesTr│      img-1_0000.nii.gz│      img-1_0001.nii.gz│      img-1_0002.nii.gz│      img-2_0000.nii.gz│      img-2_0001.nii.gz│      img-2_0002.nii.gz│      img-3_0000.nii.gz│      img-3_0001.nii.gz│      img-3_0002.nii.gz│      img-4_0000.nii.gz│      img-4_0001.nii.gz│      img-4_0002.nii.gz│      img-5_0000.nii.gz│      img-5_0001.nii.gz│      img-5_0002.nii.gz│      ...├─imagesTs│      img-10.nii.gz│      img-11.nii.gz│      ...└─labelsTrimg-1.nii.gzimg-2.nii.gzimg-3.nii.gzimg-4.nii.gzimg-5.nii.gz...
    
  2. json文件信息

    nnUNet/nnunet/dataset_conversion/utils.py里面的函数generate_dataset_json可以生成相应任务的json文件。

    { 
    "name": "MassRoadsSeg", 
    "description": "MassRoadsSeg",
    "reference": "https://www.kaggle.com/insaff/massachusetts-roads-dataset",
    "licence":'hands off!',
    "release":"1.0 06/08/2018",
    "tensorImageSize": "4D",
    "modality": { "0": "Red","1": "Green","2": "Blue"}, "labels": { "0": "background", "1": "street"}, "numTraining": 804, "numTest": 13,"training":[{"image":"./imagesTr/img-1.nii.gz","label":"./labelsTr/img-1.nii.gz"},{"image":"./imagesTr/img-2.nii.gz","label":"./labelsTr/img-2.nii.gz"},{"image":"./imagesTr/img-3.nii.gz","label":"./labelsTr/img-3.nii.gz"},{"image":"./imagesTr/img-4.nii.gz","label":"./labelsTr/img-4.nii.gz"},{"image":"./imagesTr/img-5.nii.gz","label":"./labelsTr/img-5.nii.gz"},...],"test":["./imagesTs/img-10.nii.gz","./imagesTs/img-11.nii.gz",...]}

预处理

# 只进行2d预处理,不进行3d预处理
nnUNet_plan_and_preprocess -t 120 -pl3d None

训练

nnUNet_train 2d nnUNetTrainerV2 120 0

测试

nnUNet_predict -i /root/nnUNet_raw_data_base/nnUNet_raw_data/Task120_MassRoadsSeg/imagesTs -o /root/nnUNet_trained_models/nnUNet/2d/Task120_MassRoadsSeg/nnUNetTrainerV2__nnUNetPlansv2.1/fold_0/infer -t 120 -m 2d

相关内容

热门资讯

《闪电侠》闪电侠的父亲是谁演的... 《闪电侠》闪电侠的父亲是谁演的 《闪电侠》演员确定 闪电侠的父亲扮演者是“John Wesley S...
暖心的数学公式 暖心的数学公式呃,不知道你这什么意思了,数学公式就都是那样的啊,但是笛卡儿的一个三角函数画出来是一个...
苏丹西部一难民营遭炮击 造成5... 当地时间7月6日,苏丹武装部队表示,苏丹西部北达尔富尔州首府法希尔市近郊的阿布舒克难民营当天遭遇炮击...
保险如何更好守护美好生活 今年7月8日是第13个“7·8全国保险公众宣传日”,年度主题是“爱和责任 保险让生活更美好”。近年来...
城乡一盘棋走出繁荣路 近日,国务院常务会议审议通过《农村公路条例(草案)》,对于促进乡村全面振兴意义重大。一直以来,国家积...
偷星九月天中所有人第七感 偷星九月天中所有人第七感一月 被她舔过的甜食都听她的指挥爆炸二月 控制万兽三月 控制金属四月 ...
四十而不惑的“而”是什么意思 四十而不惑的“而”是什么意思这里”而“做连词,表承接关系,在句中无实际含义,可不作具体解释。意思是说...
北京推动商务中心区国际化发展 本报北京7月6日讯(记者韩秉志)日前,2025北京CBD论坛年会在京开幕。本届论坛年会实现全面升级,...
迷失少女日记怎么样 迷失少女日记怎么样音乐很好 一听就是德国奥地利出来的 只是结局我不喜欢
怎么让自己唱歌更像原唱?(跪求... 怎么让自己唱歌更像原唱?(跪求)我想学(大海)这首歌怎么让我唱的更像张雨生的声音?更像原唱?去搜索一...