Upload folder using huggingface_hub

README.md

@@ -0,0 +1,87 @@
+---
+license: mit
+tags:
+  - image-segmentation
+  - multilabel
+  - unet
+  - pytorch
+  - medical-imaging
+library_name: transformers
+pipeline_tag: image-segmentation
+---
+
+# LN_segmentation_sweep
+
+A unet model for multilabel image segmentation trained with sliding window approach.
+
+## Model Description
+
+- **Architecture:** unet
+- **Input Channels:** 3
+- **Output Classes:** 4
+- **Base Filters:** 128
+- **Window Size:** 128
+- **Downsample Factor:** 1.0
+
+### Model-Specific Parameters
+
+
+## Training Configuration
+
+| Parameter | Value |
+|-----------|-------|
+| Batch Size | 8 |
+| Learning Rate | 7.77451918775676e-06 |
+| Weight Decay | 0.00164040349077736 |
+| Epochs | 100 |
+| Patience | 10 |
+| Dataset | GleghornLab/Semi-Automated_LN_Segmentation_10_11_2025 |
+
+## Performance Metrics
+
+No test metrics available.
+
+## Usage
+
+```python
+import numpy as np
+from model import MODEL_REGISTRY, SegmentationConfig
+
+# Load model
+config = SegmentationConfig.from_pretrained("aholk/LN_segmentation_sweep")
+model = MODEL_REGISTRY["unet"].from_pretrained("aholk/LN_segmentation_sweep")
+model.eval()
+
+# Run inference on a full image with sliding window
+image = np.random.rand(2048, 2048, 3).astype(np.float32)  # Your image here
+probs = model.predict_full_image(
+    image,
+    dim=128,
+    batch_size=16,
+    device="cuda"  # or "cpu"
+)
+# probs shape: (num_classes, H, W) with values in [0, 1]
+
+# Threshold to get binary masks
+masks = (probs > 0.5).astype(np.uint8)
+```
+
+## Training Plots
+
+![Training Loss](training_loss.png)
+![Dice Curves](dice_curves.png)
+![MCC Curves](mcc_curves.png)
+![Best Validation](best_validation_reconstruction.png)
+
+## Citation
+
+If you use this model, please cite:
+
+```bibtex
+@software{windowz_segmentation,
+  title={Multilabel Image Segmentation with Sliding Window U-Net},
+  author={Gleghorn Lab},
+  year={2025},
+  url={https://github.com/GleghornLab/ComputerVision2}
+}
+```

config.json

@@ -0,0 +1,16 @@
+{
+  "architectures": [
+    "UNetForSegmentation"
+  ],
+  "dtype": "float32",
+  "img_size": null,
+  "k": 5,
+  "model_arch": null,
+  "model_type": "segmentation",
+  "n_filts": 128,
+  "norm": null,
+  "num_channels": 3,
+  "num_classes": 4,
+  "t": 3,
+  "transformers_version": "5.3.0"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16649a1d727ddf7f3ebeb663a375f4c91d8ec9655e70f45aad83f37dc12481d1
+size 2208895776