fabianhenning/rsna_2023_abdominal_trauma_detection

RSNA 2023 Abdominal Trauma Detection 3rd place solution

Environment

pip install -r requirements.txt

Data Download

• competition data
• png data from theo

cd input
. download.sh

Train & Inference 3d Segmentation

cd script
. train_3d_segmentation.sh
. inference_3d_segmentation.sh

Create Cropped Images

python create_cropped_images.py
python create_fold_df.py

Pretrain by Image Level Label

python prepare_image_level_df.py
. pretrain.sh

Train & Inference 2.5d+3d Classification

python train_all.py
python inference_all.py

Searce Weights

python search_weights.py

Inference

https://www.kaggle.com/code/yujiariyasu/3rd-place-inf-code

CSV & Weights

Below is the finished item
csv:
https://www.kaggle.com/datasets/yujiariyasu/rsna2023-csvs
model weights:
https://www.kaggle.com/datasets/yujiariyasu/class23-maxvit-224-2segmodels-v4-pretrained
https://www.kaggle.com/datasets/yujiariyasu/class0-masked-192-2segmodels-pad0-caformer
https://www.kaggle.com/datasets/yujiariyasu/class-all-pad30-caformer
https://www.kaggle.com/datasets/yujiariyasu/class-all-pad30-maxvit-pretrained
https://www.kaggle.com/datasets/yujiariyasu/class-all-pad30-convnext-pretrained
https://www.kaggle.com/datasets/yujiariyasu/class-all-pad30-seresnext-crop2-pretrained
https://www.kaggle.com/datasets/yujiariyasu/class-all-pad30-maxvit-crop2-pretrained
https://www.kaggle.com/datasets/yujiariyasu/class1-lstm-112-20epochs-auc-v3-pretrained2-fix
https://www.kaggle.com/datasets/yujiariyasu/class23-2segmodels-v3-xcit-small
https://www.kaggle.com/datasets/yujiariyasu/class0-masked-192-2segmodels-pad0
https://www.kaggle.com/datasets/yujiariyasu/class4-pretrain-288-n25-2segmodels-25epochs-v4
https://www.kaggle.com/datasets/yujiariyasu/class4-pretrain-288-n25-2segmodels-v4
https://www.kaggle.com/datasets/yujiariyasu/class4-pretrain-288-n25-2segmodels-25epochs-v3
https://www.kaggle.com/datasets/yujiariyasu/class4-pretrain-288-n25-2segmodels-v3
https://www.kaggle.com/datasets/yujiariyasu/class0-gru-chaug-256-2segmodels-v3
https://www.kaggle.com/datasets/yujiariyasu/rsna-3dseg-resnet50-v2

GPU

NVIDIA A100 for NVLink 40GiB HBM2 x 1