RSNA

This model is for RSNA Pneumonia Detection Challenge based on Mask-RCNN and pre-trained on COCO

model_train_test.py includes train phase and test phase, it takes about 8 hours to train on 1 GPU( NVIDIA 1080Ti)

How to run?

'bash requirement.sh' #This will download Mask-RCNN and mask_rcnn_coco.h5
'python3 model_train_test.py' #Please be patient. You can also use 'nohup python3 model_train_test.py &' to run it backend.

Name		Name	Last commit message	Last commit date
Latest commit History 13 Commits
__pycache__		__pycache__
.gitignore		.gitignore
README.md		README.md
classifer_vgg.ipynb		classifer_vgg.ipynb
classifer_vgg.py		classifer_vgg.py
classifier_resnet.ipynb		classifier_resnet.ipynb
classifier_resnet.py		classifier_resnet.py
classifier_resnet50.py		classifier_resnet50.py
evaluate_classification_segmentation_detection.ipynb		evaluate_classification_segmentation_detection.ipynb
image_bbox_full.csv		image_bbox_full.csv
kernel.ipynb		kernel.ipynb
kernel.py		kernel.py
lung_opacity_overview.ipynb		lung_opacity_overview.ipynb
mdai_model.py		mdai_model.py
new_kernel.py		new_kernel.py
resnet.ckpt		resnet.ckpt
seg_mrcnn.ipynb		seg_mrcnn.ipynb
seg_mrcnn.py		seg_mrcnn.py
segmentation.ipynb		segmentation.ipynb
segmentation.py		segmentation.py
stage_1_test_labels.csv		stage_1_test_labels.csv
test_classifier_resnet.ipynb		test_classifier_resnet.ipynb
test_kernel.py		test_kernel.py
test_pretrained.ipynb		test_pretrained.ipynb

Kwongrf/RSNA