RayOoooo / RSNA2019_1st_place_solution

RSNA2019 Intracranial Hemorrhage Detection

RSNA Intracranial Hemorrhage Detection

This is the source code for the first place solution to the RSNA2019 Intracranial Hemorrhage Detection Challenge. Video with quick overview:

Solutuoin Overview

Dependencies

• opencv-python==3.4.2
• scikit-image==0.14.0
• scikit-learn==0.19.1
• scipy==1.1.0
• torch==1.1.0
• torchvision==0.2.1

code

• 2DCNN
• SequenceModel

2D CNN Classifier

Preprocessing

1. prepare csv file

python3 prepare_csv.py -root_path -train_dcm_path -test_dcm_path -save_path

root_path         kaggle data path
train_dcm_path    train dicom data path
test_dcm_path     test dicom data path
save_path         output path

2. convert dcm to png

python3 prepare_data.py -dcm_path stage_1_train_images -png_path train_png
python3 prepare_data.py -dcm_path stage_2_test_images -png_path test_png

3. train

python3 train_model.py -backbone DenseNet121_change_avg -img_size 256 -tbs 256 -vbs 128 -save_path DenseNet121_change_avg_256
python3 train_model.py -backbone DenseNet169_change_avg -img_size 256 -tbs 256 -vbs 128 -save_path DenseNet169_change_avg_256
python3 train_model.py -backbone se_resnext101_32x4d -img_size 256 -tbs 80 -vbs 40 -save_path se_resnext101_32x4d_256

4. predict

python3 predict.py -backbone DenseNet121_change_avg -img_size 256 -tbs 4 -vbs 4 -spth DenseNet121_change_avg_256
python3 predict.py -backbone DenseNet169_change_avg -img_size 256 -tbs 4 -vbs 4 -spth DenseNet169_change_avg_256
python3 predict.py -backbone se_resnext101_32x4d -img_size 256 -tbs 4 -vbs 4 -spth se_resnext101_32x4d_256

After single models training, the oof files will be saved in ./SingleModelOutput(three folders for three pipelines).

After training the sequence model, the final submission will be ./FinalSubmission/final_version/submission_tta.csv

Sequence Models

Sequence Model 1

Sequence Model 2

Path Setup

Set data path to your own in ./setting.py

download

https://drive.google.com/open?id=1qYi4k-DuOLJmyZ7uYYrnomU2U7MrYRBV

https://drive.google.com/open?id=1lJgzZoHFu6HI4JBktkGY3qMk--28IUkC

Sequence Model Training

CUDA_VISIBLE_DEVICES=0 python main.py

The final submissions are in the folder ../FinalSubmission/version2/submission_tta.csv

Final Submission

Private Leaderboard:

• 0.04383