search

MIC-DKFZ / nnUNet

Apache License 2.0
5.91k stars 1.76k forks source link

I want to create a python class for inference #2455

Closed Body123 closed 1 week ago

Body123 commented 2 months ago

I want to create a python class for inference no importance if it is long can any one help me (I want to create the usual pytorch class)

sten2lu commented 2 months ago

Hi @Body123 There is already a Python class for inference in nnU-Net. See here for an example: https://github.com/MIC-DKFZ/nnUNet/tree/master/nnunetv2/inference

Best regards, Carsten