├── LICENSE ├── README.md ├── caid_models.py ├── data_loader.py ├── dataset ├── Xray14_train_official.txt └── Xray14_val_official.txt ├── images ├── Distribution.png ├── Method.png ├── feature_reuse.png └── w_wo.png ├── main_CAiD_moco.py ├── requirements.txt ├── segmentation_models_pytorch ├── __init__.py ├── __version__.py ├── base │ ├── __init__.py │ ├── heads.py │ ├── initialization.py │ ├── model.py │ └── modules.py ├── deeplabv3 │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── encoders │ ├── __init__.py │ ├── _base.py │ ├── _preprocessing.py │ ├── _utils.py │ ├── densenet.py │ ├── dpn.py │ ├── efficientnet.py │ ├── inceptionresnetv2.py │ ├── inceptionv4.py │ ├── mobilenet.py │ ├── resnet.py │ ├── senet.py │ ├── timm_efficientnet.py │ ├── timm_regnet.py │ ├── timm_res2net.py │ ├── timm_resnest.py │ ├── timm_sknet.py │ ├── vgg.py │ └── xception.py ├── fpn │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── linknet │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── losses │ ├── __init__.py │ ├── _functional.py │ ├── constants.py │ ├── dice.py │ ├── focal.py │ ├── jaccard.py │ ├── lovasz.py │ ├── soft_bce.py │ └── soft_ce.py ├── manet │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── pan │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── pspnet │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── transskipunet │ ├── __init__.py │ ├── decoder.py │ ├── model.py │ └── transformer.py ├── unet │ ├── __init__.py │ ├── decoder.py │ └── model.py ├── unetplusplus │ ├── __init__.py │ ├── decoder.py │ └── model.py └── utils │ ├── __init__.py │ ├── base.py │ ├── functional.py │ ├── losses.py │ ├── meter.py │ ├── metrics.py │ └── train.py ├── trainer.py ├── transformation.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/README.md -------------------------------------------------------------------------------- /caid_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/caid_models.py -------------------------------------------------------------------------------- /data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/data_loader.py -------------------------------------------------------------------------------- /dataset/Xray14_train_official.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/dataset/Xray14_train_official.txt -------------------------------------------------------------------------------- /dataset/Xray14_val_official.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/dataset/Xray14_val_official.txt -------------------------------------------------------------------------------- /images/Distribution.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/images/Distribution.png -------------------------------------------------------------------------------- /images/Method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/images/Method.png -------------------------------------------------------------------------------- /images/feature_reuse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/images/feature_reuse.png -------------------------------------------------------------------------------- /images/w_wo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/images/w_wo.png -------------------------------------------------------------------------------- /main_CAiD_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/main_CAiD_moco.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/requirements.txt -------------------------------------------------------------------------------- /segmentation_models_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/__version__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/base/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/base/heads.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/base/initialization.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/base/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/base/modules.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/deeplabv3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/deeplabv3/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/deeplabv3/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/deeplabv3/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/deeplabv3/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/deeplabv3/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/_base.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/_preprocessing.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/_utils.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/densenet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/dpn.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/efficientnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/inceptionresnetv2.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/inceptionv4.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/mobilenet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/resnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/senet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/timm_efficientnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/timm_regnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_res2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/timm_res2net.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/timm_resnest.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_sknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/timm_sknet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/vgg.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/encoders/xception.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/fpn/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import FPN -------------------------------------------------------------------------------- /segmentation_models_pytorch/fpn/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/fpn/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/fpn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/fpn/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/linknet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import Linknet -------------------------------------------------------------------------------- /segmentation_models_pytorch/linknet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/linknet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/linknet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/linknet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/_functional.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/constants.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/dice.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/focal.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/jaccard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/jaccard.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/lovasz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/lovasz.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/soft_bce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/soft_bce.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/soft_ce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/losses/soft_ce.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/manet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import MAnet 2 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/manet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/manet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/manet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/manet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/pan/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import PAN -------------------------------------------------------------------------------- /segmentation_models_pytorch/pan/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/pan/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/pan/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/pan/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/pspnet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import PSPNet -------------------------------------------------------------------------------- /segmentation_models_pytorch/pspnet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/pspnet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/pspnet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/pspnet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/transskipunet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import TrSkipUnet -------------------------------------------------------------------------------- /segmentation_models_pytorch/transskipunet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/transskipunet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/transskipunet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/transskipunet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/transskipunet/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/transskipunet/transformer.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/unet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import Unet -------------------------------------------------------------------------------- /segmentation_models_pytorch/unet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/unet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/unet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/unet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/unetplusplus/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/unetplusplus/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/unetplusplus/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/unetplusplus/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/unetplusplus/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/unetplusplus/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/base.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/functional.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/losses.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/meter.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/metrics.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/segmentation_models_pytorch/utils/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/trainer.py -------------------------------------------------------------------------------- /transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/transformation.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlianglab/CAiD/HEAD/utils.py --------------------------------------------------------------------------------