pytorch image-segmentation license:apache-2.0 region:us
---
license: apache-2.0
pipeline_tag: image-segmentation
library_name: pytorch
---

This model consists of a nnU-Net trained on the GelGenie training dataset in December 2023.  It is highly robust and should produce good results for a wide variety of gel types and resolutions.  The model was packaged into a torchscript file by scripting many of the data processing functions within the model itself.  As such, it is quite a large and complicated model and will require a GPU to run efficiently.  

The model was trained on the combined training and validation set (470 images) for 600 epochs and the last checkpoint extracted.

For more details on the configuration used for training, please visit https://huggingface.co/mattaq/GelGenie-nnUNet-Extended-Dec-2023 and check the config files.

For the original nnU-Net package and for more details on the nnU-Net model itself, please check https://github.com/MIC-DKFZ/nnUNet.
