- 博客(3)
- 收藏
- 关注
RSS订阅原创 nnUNetv2使用自己数据集出现的问题nnUNetv2(二)
在nnunetv2.training.nnUNetTrainer.nnUNetTrainer的initialize函数进行初始化，根据对应的nnUNetPlans.json，如果使用3d_fullres的nnUNetv2，对应的结构为下图所示。作者使用的是dynamic_network_architectures中的unet模型，作者通过pydoc.locate加载需要的network_class_name关键字对应的类，然后将arch_kwargs中对应的模型结构作为参数初始化模型的结构。
2024-07-03 15:59:06
252
原创 nnUNetv2使用自己数据集出现的问题
处理数据之后，就可以对模型进行训练，训练指令如下。DATASET_NAME_OR_ID就是我们的数据集id(501,502)，UNET_CONFIGURATION FOLD就是使用nnunetv2提供的集中Unet配置进行训练，分别有2d、3d_fullres、3d_lowres、 3d_cascade_lowres。也就是说每个nii.gz的文件命名必须和训练集的文件命名一致，同时文件名字的最后需要有一个4位数的标识符，表示是哪一个模态的，和dataset.json中的channel_name设置一致。
2024-07-03 10:07:48
575
2
原创 python 运行出错，view改变tensor size出现错误
输入Tensor的维度出现问题。view需要tensor在子空间(地址)连续，根据提示有两种解决方法。
2023-10-02 10:38:27
57
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人