nnU-Net - ディープ ラーニング セグメンテーション メソッド ( https://github.com/MIC-DKFZ/nnUNet ) - を取得して、トレーニング データセットを使用してモデルのトレーニングを完了した後に推論を実行しようとしてnnUNetTrainerV2_DDPいます。
nnUNet_change_trainer_classモデルが保存されているフォルダーで実行したところ、フォルダー内でcheckpoint.modelおよびmodal_bestpkl ファイルが更新されていることがわかりますが、テスト セットで予測を開始した後、次のエラーが発生します。
AssertionError: model output folder not found. Expected: E:\Pieterjan\nnUNet_project\nnUNet_trained_models\nnUNet\3d_fullres\Task101_Lung\nnUNetTrainerV2__nnUNetPlansv2.1
nnUNetTrainerV2__nnUNetPlansv2.1 1 つの GPU のみでトレーニングされたモデルを使用する場合、推論を開始するときに使用されるデフォルトのようですが、2 つの GPU を使用しました (したがって_DDP trainer、nnU-Net によって提供される を使用しました)。また、単純にフォルダーの名前を変更しようとしましたが、問題は解決しません。
nnUNetTrainerV2_DDP__nnUNetPlansv2.1テスト データセットでの推論に使用する nnU-Net をセットアップするにはどうすればよいですか?
事前に乾杯！