关于在阅读nnUNet代码中的一些小细节的记录(三)

1.推理阶段:nnUNet/nnunet/training/model_restore.py的60行的recursive_find_python_class函数的参数current_module。

改为当前的network_training目录的位置。如下:

current_module="AIseg.core_prepare_result.nnunet.training.network_training"

2.推理阶段:nnUNet/nnunet/training/network_training/nnUNetTrainer.py的434行的recursive_find_python_class函数的 参数current_module。

改为当前的preprocessing目录的位置。如下

current_module="AIseg.core_prepare_result.nnunet.preprocessing"

前面加入:

from AIseg.core_prepare_result import nnunet

 

你可能感兴趣的:(python,深度学习)