|
--- |
|
license: apache-2.0 |
|
pipeline_tag: image-segmentation |
|
library_name: pytorch |
|
--- |
|
|
|
This model consists of an SMP-UNet trained on the LSDB dataset in December 2023 for GelGenie. It should only be used for low-resolution images. |
|
|
|
The model was trained on the training set (53 images) for 600 epochs and the best checkpoint extracted based on the validation dice score. |
|
|
|
For more details on the configuration used for training, please visit https://huggingface.co/mattaq/GelGenie-LowRes-Dec-2023 and check the config.toml file. Our codebase is fully open-sourced and is available here: https://github.com/mattaq31/GelGenie. |
|
|
|
|
