├── LICENSE ├── README.md ├── configs ├── irseg.json └── pst900.json ├── evaluate_PST900.py ├── evaluate_irseg.py ├── image ├── cainet.png ├── result_mfnet.jpg └── result_pst.jpg ├── requirements.txt ├── toolbox ├── Ranger.py ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── log.cpython-36.pyc │ ├── log.cpython-37.pyc │ ├── loss.cpython-36.pyc │ ├── loss.cpython-37.pyc │ ├── metrics.cpython-36.pyc │ ├── metrics.cpython-37.pyc │ ├── utils.cpython-36.pyc │ └── utils.cpython-37.pyc ├── datasets │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── augmentations.cpython-36.pyc │ │ ├── augmentations.cpython-37.pyc │ │ ├── irseg.cpython-36.pyc │ │ ├── irseg.cpython-37.pyc │ │ ├── pst900.cpython-36.pyc │ │ └── pst900.cpython-37.pyc │ ├── augmentations.py │ ├── irseg.py │ └── pst900.py ├── log.py ├── loss.py ├── losses │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── loss.cpython-36.pyc │ │ ├── loss.cpython-37.pyc │ │ ├── lovasz_losses.cpython-36.pyc │ │ └── lovasz_losses.cpython-37.pyc │ ├── loss.py │ └── lovasz_losses.py ├── metrics.py ├── models │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ └── cainet.cpython-37.pyc │ ├── backbone │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-36.pyc │ │ │ ├── __init__.cpython-37.pyc │ │ │ └── mobilenetv2.cpython-37.pyc │ │ └── mobilenetv2.py │ └── cainet.py ├── optimizer │ ├── SGD.py │ └── __pycache__ │ │ ├── SGD.cpython-36.pyc │ │ └── SGD.cpython-37.pyc ├── ranger │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── ranger.cpython-36.pyc │ │ ├── ranger.cpython-37.pyc │ │ ├── ranger913A.cpython-36.pyc │ │ ├── ranger913A.cpython-37.pyc │ │ ├── rangerqh.cpython-36.pyc │ │ └── rangerqh.cpython-37.pyc │ ├── ranger.py │ ├── ranger913A.py │ └── rangerqh.py └── utils.py ├── train_irseg.py └── train_pst900.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/README.md -------------------------------------------------------------------------------- /configs/irseg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/configs/irseg.json -------------------------------------------------------------------------------- /configs/pst900.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/configs/pst900.json -------------------------------------------------------------------------------- /evaluate_PST900.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/evaluate_PST900.py -------------------------------------------------------------------------------- /evaluate_irseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/evaluate_irseg.py -------------------------------------------------------------------------------- /image/cainet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/image/cainet.png -------------------------------------------------------------------------------- /image/result_mfnet.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/image/result_mfnet.jpg -------------------------------------------------------------------------------- /image/result_pst.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/image/result_pst.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/requirements.txt -------------------------------------------------------------------------------- /toolbox/Ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/Ranger.py -------------------------------------------------------------------------------- /toolbox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__init__.py -------------------------------------------------------------------------------- /toolbox/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/log.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/log.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/log.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/log.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/loss.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/metrics.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/metrics.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/augmentations.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/augmentations.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/augmentations.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/augmentations.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/irseg.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/irseg.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/irseg.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/irseg.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/pst900.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/pst900.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/datasets/__pycache__/pst900.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/__pycache__/pst900.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/datasets/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/augmentations.py -------------------------------------------------------------------------------- /toolbox/datasets/irseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/irseg.py -------------------------------------------------------------------------------- /toolbox/datasets/pst900.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/datasets/pst900.py -------------------------------------------------------------------------------- /toolbox/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/log.py -------------------------------------------------------------------------------- /toolbox/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/loss.py -------------------------------------------------------------------------------- /toolbox/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from .loss import * 2 | -------------------------------------------------------------------------------- /toolbox/losses/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/losses/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/losses/__pycache__/loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/__pycache__/loss.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/losses/__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/losses/__pycache__/lovasz_losses.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/__pycache__/lovasz_losses.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/losses/__pycache__/lovasz_losses.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/__pycache__/lovasz_losses.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/loss.py -------------------------------------------------------------------------------- /toolbox/losses/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/losses/lovasz_losses.py -------------------------------------------------------------------------------- /toolbox/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/metrics.py -------------------------------------------------------------------------------- /toolbox/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /toolbox/models/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/models/__pycache__/cainet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/__pycache__/cainet.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/models/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /toolbox/models/backbone/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/backbone/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/models/backbone/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/backbone/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/models/backbone/__pycache__/mobilenetv2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/backbone/__pycache__/mobilenetv2.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/models/backbone/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/backbone/mobilenetv2.py -------------------------------------------------------------------------------- /toolbox/models/cainet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/models/cainet.py -------------------------------------------------------------------------------- /toolbox/optimizer/SGD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/optimizer/SGD.py -------------------------------------------------------------------------------- /toolbox/optimizer/__pycache__/SGD.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/optimizer/__pycache__/SGD.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/optimizer/__pycache__/SGD.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/optimizer/__pycache__/SGD.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__init__.py -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/ranger.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/ranger.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/ranger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/ranger.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/ranger913A.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/ranger913A.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/ranger913A.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/ranger913A.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/rangerqh.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/rangerqh.cpython-36.pyc -------------------------------------------------------------------------------- /toolbox/ranger/__pycache__/rangerqh.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/__pycache__/rangerqh.cpython-37.pyc -------------------------------------------------------------------------------- /toolbox/ranger/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/ranger.py -------------------------------------------------------------------------------- /toolbox/ranger/ranger913A.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/ranger913A.py -------------------------------------------------------------------------------- /toolbox/ranger/rangerqh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/ranger/rangerqh.py -------------------------------------------------------------------------------- /toolbox/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/toolbox/utils.py -------------------------------------------------------------------------------- /train_irseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/train_irseg.py -------------------------------------------------------------------------------- /train_pst900.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YingLv1106/CAINet/HEAD/train_pst900.py --------------------------------------------------------------------------------