u014776464的博客

nnunet修改batchsize、patchsize本文只是为了自己平时训练方便而做的笔记，不保证一定正确，但这样修改的确能正常使用。1、首先使用官方的数据集处理命令，对数据集进行处理2、使用如下代码修改batchszie、patchsizeimport picklefrom batchgenerators.utilities.file_and_folder_operations import *import numpy as np#源plans文件的路径，基于该文件进行修改,这个保存b

使用pytorch cuda11 cudnn8.02 Docker搭建nnUNet训练环境本文目的是加速训练过程，平时使用pytorch1.6 cuda10.2训练一个epoch将近400秒，搭建好docker环境后，使用cuda11、cudnn8.0.2 的环境，一个epoch的时间只需要160秒左右。加速非常明显。问题来源于：https://github.com/MIC-DKFZ/nnUNet/issues/292项目主仓库 ：https://github.com/MIC-DKFZ/nnUNet

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档文章目录前言一、nnUnet安装二、具体步骤1.准备好你自己的数据集参考前言最近在学习nnUnet这个框架，想用在自己的数据集上，所以就github来一番，终于找到了一些简单的创建数据集的方法。一、nnUnet安装nnUNet官方地址如下：https://github.com/MIC-DKFZ/nnUNet安装可以参考如下博客：（四：2020.07.28）nnUNet最舒服的训练教程（让我的奶奶也会用nnUNet（上））（9