Skip to content
8th place solution for SIIM-ACR Pneumothorax Segmentation competition on Kaggle
Python Shell
Branch: master
Clone or download
Ian Pan
Ian Pan update README
Latest commit ba5efc3 Sep 10, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
etl initial commit Sep 9, 2019
segment initial commit Sep 9, 2019
submit initial commit Sep 9, 2019 update README Sep 10, 2019
directory_structure.txt initial commit Sep 9, 2019 update entry_points Sep 9, 2019 initial commit Sep 9, 2019 initial commit Sep 9, 2019
stage1test.txt initial commit Sep 9, 2019

Kaggle SIIM-ACR Pneumothorax Segmentation (#8/1475)


  • Ubuntu 16.04 LTS
  • 64 GB RAM / 2 TB HDD
  • 1x NVIDIA Titan V100 32GB
  • 1x Titan V 12GB


  • Python 3.7.4
  • CUDA 10.0
  • cuDNN 7.6
  • PyTorch 1.1

Model Checkpoints

Download from Google Drive:

pip install gdown

Models should be unzipped into ./segment/checkpoints/ in order to run code as is. There should be 3 folders:



See for reproducing results. Relative filepaths and directories are used, so the code should work as is.

Note that TRAIN_V100 and TRAIN_DEEPLABXY models require V100 32GB GPUs to train with the current configurations. If you wish to train these models on a lower capacity GPU, I suggest using the following flag options:

--grad-accum 8 --batch-size 2 or --grad-accum 16 --batch-size 1

Model performance is not guaranteed to be the same with these modifications.

You can’t perform that action at this time.