Downloads · 30 days
10
5% of all-time downloads
brain-synthesizer
huzey/brain-synthesizer
brain-synthesizer is a machine learning model from huzey. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Pre-trained model for NSD brain encoding.
All-time downloads
187
Public
Repo size
17.9 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth17.9 GB · 100%
At a glance
- License
- mit
- Model type
- nsd
- Access
- Public
- Created
- Mar 24, 2026
- Updated
- Mar 25, 2026
- SHA
- 274fbb83
Model card
From the Hugging Face model README
NSD Brain Encoding Model (Inference)
Pre-trained model for NSD brain encoding.
from PIL import Image
from transformers import AutoImageProcessor, AutoModel
image_processor = AutoImageProcessor.from_pretrained("facebook/dinov2-large")
image = Image.open("example.jpg").convert("RGB")
pixel_values = image_processor(images=image, return_tensors="pt")["pixel_values"]
## whole brain model, including non visual brain
model = AutoModel.from_pretrained(
"huzey/brain-synthesizer",
trust_remote_code=True,
subject="subj01", # options: subj01, subj02, subj05, subj07
)
output = model(pixel_values)
print(output.shape) # torch.Size([1, 327684]), vertices in fsaverage, first half is left hemisphere, second half right hemisphere
Subject options
subject: any subject ID with matching checkpoint files in the repo (for example,subj01,subj02,subj05,subj07).partis fixed towhole_brain: outputs all fsaverage 327684 vertices, first half is left hemisphere, second half is right hemisphere.
- Type
- nsd
- License
- mit
- Created
- Mar 24, 2026
- Updated
- Mar 25, 2026