Skip to content
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

Preparing data and train model to detect/segment spinal cord #8

Open
3 of 6 tasks
jcohenadad opened this issue Dec 7, 2021 · 6 comments
Open
3 of 6 tasks

Preparing data and train model to detect/segment spinal cord #8

jcohenadad opened this issue Dec 7, 2021 · 6 comments

Comments

@jcohenadad
Copy link
Member

jcohenadad commented Dec 7, 2021

  • Contact @andreanne-lemay to understand where the code for findcord_tumor is in https://ivadomed.org
  • Develop code to prepare data for training SC detection. This step will depend on the state of the code that @andreanne-lemay used
  • Generate preprocessed data for training cord segmentation model.
  • Package this data + version.
  • Train model. Important information: in order to avoid having big blobs far away from the cord at inference (eg caused by artifacts, OOD) there is a feature in the postprocessing methods of ivadomed that can select the biggest object:

https://github.com/ivadomed/ivadomed/blob/8fd599a07e4a20e4a4ca8fdc676c31f9a41df438/ivadomed/postprocessing.py#L102

  • Package model as new repository. Name: model_segment_sc_mp2rage
@uzaymacar
Copy link
Contributor

Hello @charleygros, Andreanne mentioned that you might have the code for training the YOLO model for SC detection. If so, can you please refer me to the code? Thanks in advance!

@uzaymacar
Copy link
Contributor

uzaymacar commented Dec 10, 2021

Also @andreanne-lemay mentioned that if we follow the config file for the cascaded architecture, and if we add the path to the model which is available on GH, it would work and crop around the SC. We'll just have to test this.

@jcohenadad
Copy link
Member Author

@andreanne-lemay @charleygros do you have any updates for us? this project is becoming quite high priority-- many thanks!

@jcohenadad jcohenadad changed the title Preparing data to train YOLO spinal cord detection Preparing data to detect spinal cord Dec 13, 2021
@jcohenadad jcohenadad changed the title Preparing data to detect spinal cord Preparing data and train model to detect/segment spinal cord Dec 13, 2021
@charleygros
Copy link
Member

@charleygros do you have any updates for us? this project is becoming quite high priority-- many thanks!

Copy that, will try to help today

@jcohenadad
Copy link
Member Author

@charleygros do you have any updates for us? this project is becoming quite high priority-- many thanks!

Copy that, will try to help today

quick reply until @uzaymacar gives more context: @uzaymacar and i had a quick chat with @andreanne-lemay who clarified a lot of things-- we've updated most of the related issues to add more context/information. So we're good for now 👍

@jcohenadad
Copy link
Member Author

Following up on #8 (comment), here is the context:

  • Andreanne/Charley did not use YOLO for the tumor segmentation. Instead, they used a UNet to segment the spinal cord, which was then used to create a mask for subsequent cropping.

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

No branches or pull requests

3 participants