Skip to content

Code for Recurrent Mask Refinement for Few-Shot Medical Image Segmentation (ICCV 2021).

Notifications You must be signed in to change notification settings

uci-cbcl/RP-Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

License CC BY-NC-SA 4.0 Python 2.7 Python 3.7

Recurrent Mask Refinement for Few-Shot Medical Image Segmentation

Steps

  1. Install any missing packages using pip or conda

  2. Preprocess each dataset using utils/preprocess

  3. Run notebooks/prepare_data_for_few_shot_learning.ipynb to generate files containing the range of each organ

Train

Test

Configure the yaml. An example is at yamls/example.yml. Change ckpt to trained model checkpoint python test.py test --yaml $PATH_TO_YAML

About

Code for Recurrent Mask Refinement for Few-Shot Medical Image Segmentation (ICCV 2021).

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages