Skip to content
Python C Cuda Shell
Branch: master
Clone or download
Latest commit 9d698bd Sep 10, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
dataSet
input
models Add files via upload Sep 8, 2019
tools
utils
README.md
common.py
compare.py
ensemble_forsub1.py
ensemble_forsub1_stage2.py
test_json.py
test_json_stage2.py
train_semi.py

README.md

SIIM-ACR-Pneumothorax-Segmentation-5th

https://www.kaggle.com/c/siim-acr-pneumothorax-segmentation/leaderboard

Dependencies

  • python==3.6
  • torch==1.0+
  • torchvision==0.3+

Solution

https://www.kaggle.com/c/siim-acr-pneumothorax-segmentation/discussion/107603#latest-620540

prepare data

competition dataset

./input/dicom-images-train
./input/dicom-images-test

NIH data

download NIH dataset
https://www.kaggle.com/nih-chest-xrays/data
common.py: EXDATAPATH = './input/NIH/images/*.png'

Train

python train_semi.py --modelname="seresnext50" --fold_index=0 --batch_size=16 --lr=3e-4

Test

stage1: python test_json.py
stage2: python test_json_stage2.py

ensemble

stage1: python ensemble_forsub1.py
stage2: python ensemble_forsub1_stage2.py

You can’t perform that action at this time.