Date: April 12th, 2021
This repo includes the brain extraction tool (DeepBet v1.0) for skull-stripping the nonhuman primate images. We also include brain masks of 136 macaque monkeys (20 sites) from PRIME-DE. The tool is constructed using a convolutional network - UNet model, initially trained on a human sample and updated with macaque data.
In this repo, we also include the outputs from other tools (AFNI, FSL, FreeSurfer, ANTS) - a glance of the performance for different pipelines
The docker image has been uploaded onto DockerHub, download it by using the following command
docker pull sandywangrest/deepbet:1.0
For the usage of this image, run
docker run sandywangrest/deepbet
~5GB hard disk space for whole docker image, including pytorch (~4GB), nibabel, scipy (~188MB), 12 U-Net models (356MB) and our code (44KB)
python3, numpy, pytorch, nibabel, scipy
python3 /path_to_the_code/muSkullStrip.py -in /path_to_the_data/input_t1.nii.gz -model /path_to_the_model/selected_model.model -out /path_to_the_output_directory
Output: *_pre_mask.nii.gz
python3 /path_to_the_code/trainSs_UNet.py -trt1w /directory_of_the_training_images -trmsk /directory_of_the_training_image_masks -out /output_directory -vt1w /directory_of_the_validation_images -vmsk /directory_of_the_validation_image_masks -init /initial_model_to_start_with
Note: Our macaque model was a transfer-learning model using a human dataset as the 'initial model' (-init option). You can use the model we provided to custimize the model for your own dataset (even across species).
The trained models can be used in prediction (muSkullStrip.py -model) or model-updating (trainSs_Unet.py -init)
-
Site-All-T-epoch_36.model: Trained on 12 macaques across 6 sites (2 macaques per site) from PRIME-DE. Six sites include newcastle, ucdavis, oxford, ion, ecnu-chen, and sbri.
-
Site-All-T-epoch_36_update_with_Site_6_plus_7-epoch_09.model: Trained on 19 macaques across 13 sites from PRIME-DE (12 macaques across 6 sites used in the first model and 7 macaques across 7 additional sites) Seven sites include NIMH, ecnu-k, nin, rockefeller, uwo, mountsinai-S, and lyon.
-
Site-All-T-epoch_36_update_with_Site_*.model: Site-specific model for NIMH, ecnu-k, nin, rockefeller, uwo, mountsinai-S, and lyon.
-
Site-All-T-epoch_36_update_with_Site_Pigs_09.model: The pig model - Trained on 12 macaques and updated with the pig data (N=3).
Manually edited brain masks for transfer-learning training (12 macaque monkeys from 6 sites, 2 per site)
Manually edited brain masks for model-updating training (7 macaque monkeys from 7 sites, 1 per site)
