Skip to content

MIPT-Oulu/RobustCartilageSegmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

rocaseg - Robust Cartilage Segmentation from MRI

Source code for Panfilov et al. "Improving Robustness of Deep Learning Based Knee MRI Segmentation: Mixup and Adversarial Domain Adaptation", https://arxiv.org/abs/1908.04126v3.

Overview

Important!

The camera-ready version contained a bug in Dice score computation for tibial cartilage on Dataset C. Please, refer to the arXiv version for the corrected values - https://arxiv.org/abs/1908.04126v3.

Description

  1. To reproduce the experiments from the article one needs to have access to OAI iMorphics, OKOA, and MAKNEE datasets.

  2. Download code from this repository.

  3. Create a fresh Conda environment using environment.yml. Install the downloaded code as a Python module.

  4. datasets/prepare_dataset_... files show how the raw data is converted into the format supported by the training and the inference pipelines.

  5. The structure of the project has to be as follows:

    ./project/
        | ./data_raw/  # raw scans and annotations
             | ./OAI_iMorphics_scans/
             | ./OAI_iMorphics_annotations/
             | ./OKOA/
             | ./MAKNEE/
        | ./data/  # preprocessed scans and annotations
        | ./src/ (this repository)
        | ./results/  # models' weights, intermediate and final results 
             | ./0_baseline/
                  | ./weights/
                  | ...
             | ./1_mixup/
             | ./2_mixup_nowd/
             | ./3_uda1/
             | ./4_uda2/
             | ./5_uda1_mixup_nowd/
    
  6. File scripts/runner.sh contains the complete description of the workflow.

  7. Statistical testing is implemented in notebooks/Statistical_tests.ipynb.

  8. Pretrained models are available at https://drive.google.com/open?id=1f-gZ2wCf55OVjgA8oXd7xttGVW5DUUcU .

Legal aspects

This code is freely available only for research purposes.

The software has not been certified as a medical device and, therefore, must not be used for diagnostic purposes.

Commercial use of the provided code and the pre-trained models is strictly prohibited, since they were developed using the medical datasets under restrictive licenses.

Cite this work

@InProceedings{Panfilov_2019_ICCV_Workshops,
  author = {Panfilov, Egor and Tiulpin, Aleksei and Klein, Stefan and Nieminen, Miika T. and Saarakkala, Simo},
  title = {Improving Robustness of Deep Learning Based Knee MRI Segmentation: Mixup and Adversarial Domain Adaptation},
  booktitle = {The IEEE International Conference on Computer Vision (ICCV) Workshops},
  month = {Oct},
  year = {2019}
}