Initial upload via tools/push_to_hf.py (architecture: ilex.models.bme_x.BMEX)

config.json

@@ -21,7 +21,7 @@
   "equinox_version": "0.13.8",
   "ilex_version": "0.0.0.dev0",
   "image_classes": "Single-channel 3D brain MRI (T1- or T2-weighted, per the variant's modality), resampled to 0.8 mm isotropic resolution per the upstream preprocessing, patched into 32^3 or 64^3 windows for inference.",
-  "intended_use": "Inference-only enhancement of a single-channel 3D brain MRI volume. Inputs are conformed to 0.8 mm isotropic, intensity-clipped/normalised, and tiled into 32^3 or 64^3 patches by the upstream sliding-window inference. Each variant's weights are tuned for a specific age range + modality (T1 vs T2). Per Sun et al. NBME 2025, the Month24 variants are the de-facto adult-applicable models. The seg head's 4-channel output (BG / WM / GM / CSF) is intended as an auxiliary signal that drives the reconstruction head's tissue-aware enhancement; downstream consumers may take softmax for probabilities and argmax for a 4-class tissue parcellation. The recon head's 1-channel output is the enhanced volume (motion-corrected, denoised, super-resolved, harmonized -- per the paper, the same network handles all four enhancement tasks).",
+  "intended_use": "Inference-only enhancement of a single-channel 3D brain MRI volume. Inputs are expected to be conformed to 0.8 mm isotropic, brain- extracted, and intensity-normalised via histogram-matching against an age-specific template (the upstream's BME_X_enhanced.py pipeline performs all three steps), then tiled into 32^3 patches with sliding-window inference at 0.85 overlap. Each variant's weights are tuned for a specific age range + modality (T1 vs T2). Per Sun et al. NBME 2025, the Month24 variants are the de-facto adult-applicable models. The seg head's 4-channel pre-softmax relu output (BG / WM / GM / CSF; apply softmax for probabilities) is an auxiliary signal that drives the reconstruction head's tissue-aware enhancement; the recon head's 1-channel output is the enhanced volume (motion- corrected, denoised, super-resolved, harmonized in one forward). CRITICAL CAVEAT (v0) -- without the upstream's histogram-matching preprocessing, the recon head produces all-zero output for at least eight of the fifteen released variants, including both adult-applicable ones (month24-t1, month24-t2). The seg head's argmax also collapses to all-background on un-preprocessed inputs. This is a property of the upstream's trained weights, not a port bug -- PyTorch and JAX agree to ~5e-5 max abs in all measured cases. See DEVNOTES.md \"Recon head usability\" for the per-variant breakdown. Until the BMEX image-domain pipeline lands in ilex.models.bme_x.pipeline, treat this port as the seg head + a partial recon head; do not consume recon directly without running the upstream's BME_X.py preprocessing first.",
   "jax_version": "0.10.0",
   "network_data_format": {
     "inputs": {},

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ded835d3786f504b2967b882449125dbeb7914f13259846fac0efab27edbe33
+oid sha256:ecd66e526a061b0fbb64b9e8d5c0a74a6cb77ab7abb4c953ad0cc8ec1058f839
 size 131856188