Skip to content

Predict any scan and export detections - #24

Merged
youssefassis merged 1 commit into
mainfrom
feat/predict-scans
Sep 29, 2026
Merged

youssefassis merged 1 commit into
mainfrom
feat/predict-scans

Conversation

@youssefassis

Copy link
Copy Markdown
Owner

predict.py can now predict any scan or folder of scans, besides the test patients:

uv run python scripts/predict.py train_dir=Data_dir/0Work/my_training input=scan.nii.gz output_dir=results

Scans are skull-stripped and normalized like the training data. For every case it writes the probability map and the detections (center and radius in mm, size, probability) as CSV and as .fcsv markups for 3D Slicer.

predict.py takes input=<scan or folder> besides the test patients of
the split. Input scans are skull-stripped and normalized like the
training data (skull_strip, the training's normalization). For every
case it writes the probability map and the detections: center and
radius in mm, size and probability, as CSV and as .fcsv markups for
3D Slicer. remove_skull.py and predict.py share skull_strip and
normalize_volume.
@youssefassis
youssefassis merged commit 6f00bb2 into main Sep 29, 2026
3 checks passed
@youssefassis
youssefassis deleted the feat/predict-scans branch September 29, 2026 18:25
youssefassis added a commit that referenced this pull request Sep 30, 2026
Predict any scan and export detections
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant