cap5610-project

CAP 5610 Project - Automatic segmentation of multiple structures from short-axis cardiac MRI

Data Preprocessing

• dataset_utils.py

Data Offline Augmentation

• ds_augmentation.py

Model Architecture

• network_2d.py
• network_2dpp.py
• network_3d.py
• model.py

Data Generators

• data.py
• data_hl.py
• data_3d.py
• data_augmented.py
• data_augmented_3d.py

Models Training

• train_hl.py
• train_2dn.py
• train_2dnpp.py
• train_3dn.py

Models Prediction

• pred_2dn.py
• pred_2dnpp.py
• pred_3dn.py

Models Evaluation

• evaluation.py

Weighted Average

• weight-hl.py
• weight-framework.py

Models Test Set Prediction

• test_hl.py
• test_2dn.py
• test_2dnpp.py
• test_3dn.py
• test_set_prediction.py