liamchalcroft/synthstroke-qatlas

qATLAS

qATLAS model trained on synthetic qMRI parameter methods predicted from ATLAS T1w.

Model Details

• Name: qATLAS
• Classes: 0 (Background), 1 (Stroke)
• Patch Size: 192³
• Voxel Spacing: 1mm³
• Input Channels: 1

Usage

Loading from Hugging Face Hub

import torch
from synthstroke_model import SynthStrokeModel

# Load the model from Hugging Face Hub
model = SynthStrokeModel.from_pretrained("liamchalcroft/synthstroke-qatlas")

# Prepare your input (example shape: batch_size=1, channels=1, H, W, D)
input_tensor = torch.randn(1, 1, 192, 192, 192)

# Get predictions (with optional TTA for improved accuracy)
predictions = model.predict_segmentation(input_tensor, use_tta=True)

# Get lesion probability map (channel 1)
lesion_probs = predictions[:, 1]  # Shape: (batch_size, H, W, D)

# Alternative: Get logits without TTA
logits = model.predict_segmentation(input_tensor, apply_softmax=False)

Citation

arXiv

@misc{chalcroft2025domainagnosticstrokelesionsegmentation,
      title={Domain-Agnostic Stroke Lesion Segmentation Using Physics-Constrained Synthetic Data}, 
      author={Liam Chalcroft and Jenny Crinion and Cathy J. Price and John Ashburner},
      year={2025},
      eprint={2412.03318},
      archivePrefix={arXiv},
      primaryClass={eess.IV},
      url={https://arxiv.org/abs/2412.03318}, 
}

License

MIT License - see the LICENSE file for details.