Skip to content

Add option to specify custom trained model for brain extraction #25

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
2 tasks
sebastientourbier opened this issue Nov 5, 2020 · 1 comment
Open
2 tasks
Labels
enhancement New feature or request little-effort

Comments

@sebastientourbier
Copy link
Member

sebastientourbier commented Nov 5, 2020

This new feature can be accomplished by:

  • Addition of new arguments --custom_ckpt_brain_localisation and --custom_ckpt_brain_localisation in pymialsrtk/parser.py(where the parser of the BIDS App is defined) to specify custom paths (prefixes) to the files of trained networks. As we are using a container image, these files should be mounted and accessible inside it. I would suggest to use to code/ folder to put these files which is already mounted.

  • Modification of docker/bidsapp/run.py and/or pymialsrtk/pipelines/anatomical/srr.py such that in_ckpt_loc for localization and in_ckpt_seg for segmentation are set to the custom paths (prefixes) specified by the new BIDS App arguments.

@sebastientourbier sebastientourbier added the enhancement New feature or request label Nov 5, 2020
@hamzake
Copy link
Member

hamzake commented Nov 9, 2020

This is incorporated in the two checkpoints to feed the node with (in_ckpt_loc for localization and in_ckpt_seg for segmentation)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request little-effort
Projects
None yet
Development

No branches or pull requests

2 participants