Pytorch example for docker submission?

Pytorch example for docker submission?  

  By: meow on July 26, 2021, 11:34 a.m.

Does somebody have an example how a pytorch model should be submitted via docker?

So far I found this example but it does not seem to match 1:1
https://github.com/DIAGNijmegen/drive-vessels-unet/tree/master/vesselSegmentor

Thanks!

Re: Pytorch example for docker submission?  

  By: kvanwijnen on July 29, 2021, 9:01 a.m.

Thank you for your message and great to hear you are participating in our challenge!

We just added a PyTorch example to the template repo, see the pytorch example branch: https://github.com/kimvwijnen/valdo-docker-template/tree/example_pytorch

You can look at the example you mentioned (https://github.com/DIAGNijmegen/drive-vessels-unet/tree/master/vesselSegmentor) as well, but please keep to the template we use in our template repo (https://github.com/kimvwijnen/valdo-docker-template/tree/main), so your Docker will run correctly in our environment. We adapted the grand-challenge template (that is used in the VesselSegmentor example) to run on our system, allow for multiple input channels (since we have multiple modalities available) and save the prediction(s) correctly etc. Hope this helps!

 Last edited by: kvanwijnen on July 29, 2021, 9:02 a.m., edited 1 time in total.