Merge pull request #708 from Pixilib/GaelO2-dev

change model to attention unet
Pixilib · Dec 28, 2023 · 90e5bed · 90e5bed
2 parents 71e7c7b + 9961f6a
commit 90e5bed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php b/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php
@@ -46,7 +46,7 @@ public function runInference(): MaskProcessingService
             'idCT' => $idCT
         ];
 
-        $inferenceResponse = $this->gaelOProcessingService->executeInference('unet_model', $inferencePayload);
+        $inferenceResponse = $this->gaelOProcessingService->executeInference('pt_seg_attentionunet', $inferencePayload);
         $maskId = $inferenceResponse['id_mask'];
         $maskProcessingService = new MaskProcessingService($this->orthancService, $this->gaelOProcessingService);
         $maskProcessingService->setMaskId($maskId);