How do I make inference?

[sergiocanar]

After training the model, I'm encountering difficulties when attempting to test the model and visualize the predictions. Specifically, after running inference on the data, Im not able to match the results you guys showed in your repository.

[bnavard]

Hello, can you please give more detail on what you do during the prediction and what data you used for it ?

[sergiocanar]

Sure! Im using Brats2017 that you guys uploaded in the repository. And after I pass one of the modalities.pt through the model I get the logits, but Im unsure to how obtain the final segmentation volume to visualize it.

[bnavard]

Which code are you using ? Can you put it here ?

[sergiocanar]

Sure! Its in my forked repo but this is how I extract the volumes from the logits
inference.txt