Commit ·
08a2e4d
1
Parent(s): 571ae9a
updated README
Browse files
README.md
CHANGED
|
@@ -7,27 +7,29 @@ tags:
|
|
| 7 |
- brain-mri
|
| 8 |
- tinygrad
|
| 9 |
- skull-stripping
|
| 10 |
-
license:
|
| 11 |
pipeline_tag: image-segmentation
|
| 12 |
-
datasets:
|
| 13 |
-
- openneuro/ds003800
|
| 14 |
---
|
| 15 |
|
| 16 |
# MindGrab (BrainChop MeshNet)
|
| 17 |
|
| 18 |
-
MindGrab is a MeshNet-based skull-stripping model from the
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
|
| 20 |
## Files
|
| 21 |
|
| 22 |
-
| File
|
| 23 |
-
|------|-------------|
|
| 24 |
-
| `model.json`
|
| 25 |
-
| `layers.json` |
|
| 26 |
-
| `model.pth`
|
| 27 |
|
| 28 |
-
Optionally, you can upload the WebGPU export (`net_mindgrab.js` + `net_mindgrab.safetensors`) from [`niivue-tinygrad`](https://github.com/spikedoanz/niivue-tinygrad) if you want to distribute browser-ready assets alongside the raw weights.
|
| 29 |
|
| 30 |
-
Optionally, you can also run the model through the official frontend at
|
|
|
|
| 31 |
|
| 32 |
|
| 33 |
## Usage
|
|
|
|
| 7 |
- brain-mri
|
| 8 |
- tinygrad
|
| 9 |
- skull-stripping
|
| 10 |
+
license: mit
|
| 11 |
pipeline_tag: image-segmentation
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# MindGrab (BrainChop MeshNet)
|
| 15 |
|
| 16 |
+
MindGrab is a MeshNet-based skull-stripping model from the
|
| 17 |
+
[BrainChop](https://github.com/neuroneural/brainchop) project. It takes 256^3
|
| 18 |
+
conformed T1 volumes and produces a binary brain mask. The checkpoint runs
|
| 19 |
+
entirely in [tinygrad](https://github.com/tinygrad/tinygrad) and powers the
|
| 20 |
+
in-browser BrainChop demos (WebGPU/WebGL).
|
| 21 |
|
| 22 |
## Files
|
| 23 |
|
| 24 |
+
| File | Description |
|
| 25 |
+
|------ |-------------|
|
| 26 |
+
| `model.json` | MeshNet architecture definition (in/out channels, kernel sizes, bias, dropout flags). |
|
| 27 |
+
| `layers.json` | Optional Layer configs |
|
| 28 |
+
| `model.pth` | FP32 PyTorch checkpoint |
|
| 29 |
|
|
|
|
| 30 |
|
| 31 |
+
Optionally, you can also run the model through the official frontend at
|
| 32 |
+
[brainchop.org](https://brainchop.org/)
|
| 33 |
|
| 34 |
|
| 35 |
## Usage
|