├── README.md ├── config.py ├── dataloader ├── data_loader.py ├── data_spliter.py ├── transform.py └── utils.py ├── dataset ├── brain_growth_alpha.zip ├── isic_attributes.zip ├── lidc_attributes.zip ├── trimap_ISIC.zip ├── trimap_LIDC.zip └── trimap_brain.zip ├── evaluate.py ├── model ├── loss_functions.py ├── loss_strategy.py ├── matting_network │ ├── cbam.py │ ├── matting_net.py │ └── resnet_block.py ├── medical_matting.py ├── metrics │ ├── compute_connectivity_error.py │ ├── compute_gradient_loss.py │ ├── compute_mse_loss.py │ ├── compute_sad_loss.py │ ├── dice_accuracy.py │ ├── generalised_energy_distance.py │ └── utils.py ├── probabilistic_unet │ ├── axis_aligned_conv_gaussian.py │ ├── encoder.py │ ├── fcomb.py │ └── prob_unet.py ├── unet │ ├── unet.py │ └── unet_blocks.py └── utils.py ├── params_brain.yaml ├── params_isic.yaml ├── params_lidc.yaml ├── requirements.txt ├── train.py └── utils ├── logger.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/config.py -------------------------------------------------------------------------------- /dataloader/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataloader/data_loader.py -------------------------------------------------------------------------------- /dataloader/data_spliter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataloader/data_spliter.py -------------------------------------------------------------------------------- /dataloader/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataloader/transform.py -------------------------------------------------------------------------------- /dataloader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataloader/utils.py -------------------------------------------------------------------------------- /dataset/brain_growth_alpha.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataset/brain_growth_alpha.zip -------------------------------------------------------------------------------- /dataset/isic_attributes.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataset/isic_attributes.zip -------------------------------------------------------------------------------- /dataset/lidc_attributes.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataset/lidc_attributes.zip -------------------------------------------------------------------------------- /dataset/trimap_ISIC.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataset/trimap_ISIC.zip -------------------------------------------------------------------------------- /dataset/trimap_LIDC.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataset/trimap_LIDC.zip -------------------------------------------------------------------------------- /dataset/trimap_brain.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/dataset/trimap_brain.zip -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/evaluate.py -------------------------------------------------------------------------------- /model/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/loss_functions.py -------------------------------------------------------------------------------- /model/loss_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/loss_strategy.py -------------------------------------------------------------------------------- /model/matting_network/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/matting_network/cbam.py -------------------------------------------------------------------------------- /model/matting_network/matting_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/matting_network/matting_net.py -------------------------------------------------------------------------------- /model/matting_network/resnet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/matting_network/resnet_block.py -------------------------------------------------------------------------------- /model/medical_matting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/medical_matting.py -------------------------------------------------------------------------------- /model/metrics/compute_connectivity_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/compute_connectivity_error.py -------------------------------------------------------------------------------- /model/metrics/compute_gradient_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/compute_gradient_loss.py -------------------------------------------------------------------------------- /model/metrics/compute_mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/compute_mse_loss.py -------------------------------------------------------------------------------- /model/metrics/compute_sad_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/compute_sad_loss.py -------------------------------------------------------------------------------- /model/metrics/dice_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/dice_accuracy.py -------------------------------------------------------------------------------- /model/metrics/generalised_energy_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/generalised_energy_distance.py -------------------------------------------------------------------------------- /model/metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/metrics/utils.py -------------------------------------------------------------------------------- /model/probabilistic_unet/axis_aligned_conv_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/probabilistic_unet/axis_aligned_conv_gaussian.py -------------------------------------------------------------------------------- /model/probabilistic_unet/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/probabilistic_unet/encoder.py -------------------------------------------------------------------------------- /model/probabilistic_unet/fcomb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/probabilistic_unet/fcomb.py -------------------------------------------------------------------------------- /model/probabilistic_unet/prob_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/probabilistic_unet/prob_unet.py -------------------------------------------------------------------------------- /model/unet/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/unet/unet.py -------------------------------------------------------------------------------- /model/unet/unet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/unet/unet_blocks.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/model/utils.py -------------------------------------------------------------------------------- /params_brain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/params_brain.yaml -------------------------------------------------------------------------------- /params_isic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/params_isic.yaml -------------------------------------------------------------------------------- /params_lidc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/params_lidc.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/train.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangsssky/MedicalMatting/HEAD/utils/utils.py --------------------------------------------------------------------------------