File size: 624 Bytes
b057fa7
 
5955964
 
b057fa7
5955964
 
 
 
 
3cc76f4
5955964
1
2
3
4
5
6
7
8
9
10
11
12
13
---
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.