# Error Using nVIDIA AIAA - Prostate Model 

**URL:** <https://discourse.slicer.org/t/error-using-nvidia-aiaa-prostate-model/20367>\
**Category:** Support\
**Tags:** segmentation, aiaa, ai\
**Created:** [October 26, 2021, 6:55pm UTC](https://discourse.slicer.org/t/error-using-nvidia-aiaa-prostate-model/20367 "2021-10-26T18:55:12Z")\
**Posts on this page:** 3\
**Page:** 1

<div class="post-metadata">

**Author:** ![mario.boccia1992](https://avatars.discourse-cdn.com/v4/letter/m/a87d85/32.png) [@mario.boccia1992](https://discourse.slicer.org/u/mario.boccia1992)\
**Post date:** [October 26, 2021, 6:55pm UTC](https://discourse.slicer.org/t/error-using-nvidia-aiaa-prostate-model/20367/1 "2021-10-26T18:55:12Z")

</div>

Dear all,

Unfortunately I’m facing issue using nVIDIA AIAA module: prostate MRI segmentation.

Running the AI I get this error:

" Traceback (most recent call last):

File “C:/Users/M/AppData/Local/NA-MIC/Slicer 4.11.20210226/NA-MIC/Extensions-29738/NvidiaAIAssistedAnnotation/lib/Slicer-4.11/qt-scripted-modules/SegmentEditorNvidiaAIAALib/SegmentEditorEffect.py”, line 410, in onClickSegmentation

extreme\_points, result\_file = self.logic.segmentation(in\_file, session\_id, model)

File “C:/Users/M/AppData/Local/NA-MIC/Slicer 4.11.20210226/NA-MIC/Extensions-29738/NvidiaAIAssistedAnnotation/lib/Slicer-4.11/qt-scripted-modules/SegmentEditorNvidiaAIAALib/SegmentEditorEffect.py”, line 1092, in segmentation

session\_id=session\_id,

File “C:\Users\M\AppData\Local\NA-MIC\Slicer 4.11.20210226\NA-MIC\Extensions-29738\NvidiaAIAssistedAnnotation\lib\Slicer-4.11\qt-scripted-modules\NvidiaAIAAClientAPI\client\_api.py”, line 389, in inference

raise AIAAException(AIAAError.SERVER\_ERROR, ‘Status: {}; Response: {}’.format(status, form))

NvidiaAIAAClientAPI.client\_api.AIAAException: (3, ‘Status: 500; Response: b’{“error”:{“message”:[],“type”:“InferenceServerException”},“success”:false}\n’’) "

Please, do you have any clues?

Thank you very much!

M.

---

<div class="post-metadata">

**Author:** ![lassoan](https://sea2.discourse-cdn.com/flex002/user_avatar/discourse.slicer.org/lassoan/32/13_2.png) [@lassoan](https://discourse.slicer.org/u/lassoan)\
**Post date:** [October 28, 2021, 4:33am UTC](https://discourse.slicer.org/t/error-using-nvidia-aiaa-prostate-model/20367/2 "2021-10-28T04:33:33Z")

</div>

You can find more detailed server logs here: [http://perklabseg.asuscomm.com:5000/logs/](http://perklabseg.asuscomm.com:5000/logs/)

```auto
[2021-10-28 04:20:36] [INFO] (aiaa.www.api.api_v1) - ===============================================================================
[2021-10-28 04:20:36] [INFO] (aiaa.www.api.api_v1) - API_v1:: /v1/inference (model: clara_pt_prostate_mri_segmentation)
[2021-10-28 04:20:36] [INFO] (aiaa.www.api.api_v1) - ===============================================================================
[2021-10-28 04:20:37] [INFO] (aiaa.inference.inference_utils) - PRE - Run Transforms
[2021-10-28 04:20:37] [INFO] (aiaa.inference.inference_utils) - PRE - Input Keys: dict_keys(['image', 'image_path', 'params', 'transform_meta'])
[2021-10-28 04:20:37] [INFO] (aiaa.inference.inference_utils) - PRE - Transform (LoadImaged): Time: 0.0883; image: (512, 512, 27)
[2021-10-28 04:20:37] [INFO] (aiaa.inference.inference_utils) - PRE - Transform (AddChanneld): Time: 0.0000; image: (1, 512, 512, 27)
[2021-10-28 04:20:37] [INFO] (aiaa.inference.inference_utils) - PRE - Transform (Spacingd): Time: 0.1367; image: (1, 161, 161, 79)
[2021-10-28 04:20:37] [INFO] (aiaa.inference.inference_utils) - PRE - Transform (ScaleIntensityRanged): Time: 0.0055; image: (1, 161, 161, 79)
[2021-10-28 04:20:37] [INFO] (aiaa.inference.triton_inference) - Input Shape: (1, 1, 161, 161, 79), Type: ndarray
[2021-10-28 04:20:37] [ERROR] (aiaa.www.api.api_v1) - unexpected shape for input 'INPUT__0' for model 'clara_pt_prostate_mri_segmentation'. Expected [-1,2,192,192,64], got [1,1,192,192,64]
[2021-10-28 04:20:37] Traceback (most recent call last):
[2021-10-28 04:20:37] File "/opt/conda/lib/python3.8/site-packages/flask/app.py", line 1950, in full_dispatch_request
[2021-10-28 04:20:37] rv = self.dispatch_request()
[2021-10-28 04:20:37] File "/opt/conda/lib/python3.8/site-packages/flask/app.py", line 1936, in dispatch_request
[2021-10-28 04:20:37] return self.view_functions[rule.endpoint](**req.view_args)
[2021-10-28 04:20:37] File "www/api/api_v1.py", line 368, in api_v1_inference
[2021-10-28 04:20:37] File "www/api/api_v1.py", line 255, in run_inference
[2021-10-28 04:20:37] File "www/api/api_v1.py", line 168, in run_infer
[2021-10-28 04:20:37] File "actions/inference_engine.py", line 80, in run
[2021-10-28 04:20:37] File "actions/inference_engine.py", line 141, in _run_inference
[2021-10-28 04:20:37] File "inference/triton_inference.py", line 74, in inference
[2021-10-28 04:20:37] File "inference/triton/inference.py", line 260, in sw_inference
[2021-10-28 04:20:37] File "/opt/conda/lib/python3.8/site-packages/tritonclient/http/ __init__.py", line 1286, in get_result
[2021-10-28 04:20:37] _raise_if_error(response)
[2021-10-28 04:20:37] File "/opt/conda/lib/python3.8/site-packages/tritonclient/http/ __init__.py", line 63, in _raise_if_error
[2021-10-28 04:20:37] raise error
[2021-10-28 04:20:37] tritonclient.utils.InferenceServerException: unexpected shape for input 'INPUT__0' for model 'clara_pt_prostate_mri_segmentation'. Expected [-1,2,192,192,64], got [1,1,192,192,64]

```

The prostate model probably fails because [it requires two input images (T2 and ADC)](https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_prostate_mri_segmentation) and most likely the AIAA client that NVidia developers created for Slicer does not support this yet.

Please submit a feature request to NVidia developers at [Issues · NVIDIA/ai-assisted-annotation-client · GitHub](https://github.com/NVIDIA/ai-assisted-annotation-client/issues) and copy the link here.

---

<div class="post-metadata">

**Author:** ![mario.boccia1992](https://avatars.discourse-cdn.com/v4/letter/m/a87d85/32.png) [@mario.boccia1992](https://discourse.slicer.org/u/mario.boccia1992)\
**Post date:** [October 28, 2021, 12:31pm UTC](https://discourse.slicer.org/t/error-using-nvidia-aiaa-prostate-model/20367/3 "2021-10-28T12:31:17Z")

</div>

Dear Andras,

I successfully created a new issue regarding this topic on the Github project.

Here you can find the link to the issue raised: [Slicer 3D - Error Using nVIDIA AIAA - Prostate Model · Issue #92 · NVIDIA/ai-assisted-annotation-client (github.com)](https://github.com/NVIDIA/ai-assisted-annotation-client/issues/92).