RSNA Intracranial Hemorrhage Detection with python

Last update: Nov 30, 2022

Overview

RSNA Intracranial Hemorrhage Detection

This is the source code for the first place solution to the RSNA2019 Intracranial Hemorrhage Detection Challenge.

Solution write up: Link.

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

2DNet
3DNet
SequenceModel

2D CNN Classifier

Pretrained models

seresnext101_256*256 [seresnext101]
densenet169_256*256 [densenet169]
densenet121_512*512 [densenet121]

Preprocessing

Prepare csv file:

download data.zip: https://drive.google.com/open?id=1buISR_b3HQDU4KeNc_DmvKTYJ1gvj5-3

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_1_test_images -png_path train_png
python3 prepare_data.py -dcm_path stage_2_test_images -png_path test_png

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

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 in ./setting.py

download

download [csv.zip]

download [feature samples]

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

Reference

If you find our work useful in your research or if you use parts of this code please consider citing our paper:

  title={A deep learning algorithm for automatic detection and classification of acute intracranial hemorrhages in head CT scans},
  author={Wang, Xiyue and Shen, Tao and Yang, Sen and Lan, Jun and Xu, Yanming and Wang, Minghui and Zhang, Jing and Han, Xiao},
  journal={NeuroImage: Clinical},
  volume={32},
  pages={102785},
  year={2021},
  publisher={Elsevier}
}

TODO

Pre-trained models
2DCNN + SeqModel end-to-end training
3DCNN training

RSNA Intracranial Hemorrhage Detection with python

Related tags

Overview

RSNA Intracranial Hemorrhage Detection

Solutuoin Overview

Dependencies

CODE

2D CNN Classifier

Pretrained models

Preprocessing

Sequence Models

Sequence Model 1

Sequence Model 2

Path Setup

download

Sequence Model Training

Final Submission

Private Leaderboard:

Reference

TODO

Owner

TResNet: High Performance GPU-Dedicated Architecture

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

《Where am I looking at? Joint Location and Orientation Estimation by Cross-View Matching》(CVPR 2020)

MVP Benchmark for Multi-View Partial Point Cloud Completion and Registration

Split your patch similarly to `git add -p` but supporting multiple buckets

An efficient PyTorch library for Global Wheat Detection using YOLOv5. The project is based on this Kaggle competition Global Wheat Detection (2021).

Repository for the paper titled: "When is BERT Multilingual? Isolating Crucial Ingredients for Cross-lingual Transfer"

Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021.

This repository compare a selfie with images from identity documents and response if the selfie match.

From this paper "SESNet: A Semantically Enhanced Siamese Network for Remote Sensing Change Detection"

Transformers based fully on MLPs

Demos of essentia classifiers hosted on replicate.ai

ImageBART: Bidirectional Context with Multinomial Diffusion for Autoregressive Image Synthesis

A robust camera and Lidar fusion based velocity estimator to undistort the pointcloud.

This project deals with the detection of skin lesions within the ISICs dataset using YOLOv3 Object Detection with Darknet.

TF2 implementation of knowledge distillation using the "function matching" hypothesis from the paper Knowledge distillation: A good teacher is patient and consistent by Beyer et al.

Codecov coverage standard for Python

This code provides various models combining dilated convolutions with residual networks

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder

Fortuitous Forgetting in Connectionist Networks