TotalSegmentator icon indicating copy to clipboard operation
TotalSegmentator copied to clipboard

Published 20 hours ago • verified publisher icon wasserth

ONNX Export

Open enochilluminant opened this issue 1 year ago • 2 comments

Is there currently a way to export the final trained model into ONNX or some similar format? This would be to deploy the model in a standalone c++ or dotnet application. I can provide more information if needed!

enochilluminant avatar Feb 08 '24 21:02 enochilluminant

If nnU-Net suppports ONNX, then TotalSegmentator can also do this, because TotalSegmentator is using a nnU-Net model. As far as I know this is not supported by nnU-Net so far (https://github.com/MIC-DKFZ/nnUNet).

wasserth avatar Feb 09 '24 08:02 wasserth

I also recently trained my own model with nnunet, and I want to deploy this model into production, have you implemented it yet? Can we communicate?

shifuxiao avatar May 07 '24 08:05 shifuxiao