├── README.md ├── data ├── Birds.py ├── CamVid.py ├── CityScapes.py ├── LiTS.py └── SkinLesion.py ├── dataset └── camvid │ ├── test │ ├── 0001TP_008550.png │ ├── 0001TP_008580.png │ ├── 0001TP_008610.png │ ├── 0001TP_008640.png │ ├── 0001TP_008670.png │ └── 0001TP_008700.png │ ├── testannot │ ├── 0001TP_008550.png │ ├── 0001TP_008580.png │ ├── 0001TP_008610.png │ ├── 0001TP_008640.png │ ├── 0001TP_008670.png │ └── 0001TP_008700.png │ ├── train │ ├── 0001TP_006690.png │ ├── 0001TP_006720.png │ ├── 0001TP_006750.png │ ├── 0001TP_006780.png │ ├── 0001TP_006810.png │ └── 0001TP_006840.png │ ├── trainannot │ ├── 0001TP_006690.png │ ├── 0001TP_006720.png │ ├── 0001TP_006750.png │ ├── 0001TP_006780.png │ ├── 0001TP_006810.png │ └── 0001TP_006840.png │ ├── val │ ├── 0016E5_07959.png │ ├── 0016E5_07961.png │ ├── 0016E5_07963.png │ ├── 0016E5_07965.png │ ├── 0016E5_07967.png │ └── 0016E5_07969.png │ └── valannot │ ├── 0016E5_07959.png │ ├── 0016E5_07961.png │ ├── 0016E5_07963.png │ ├── 0016E5_07965.png │ ├── 0016E5_07967.png │ └── 0016E5_07969.png ├── figures ├── method.jpg └── overview.jpg ├── main.py ├── models ├── attonimage │ └── encoder_channelatt_img.py ├── base │ ├── __init__.py │ ├── _base.py │ ├── _preprocessing.py │ ├── _utils.py │ ├── heads.py │ ├── initialization.py │ ├── model.py │ └── modules.py ├── deeplabv3 │ ├── deeplabv3_decoder.py │ └── deeplabv3_model.py ├── fpn │ ├── fpn_decoder.py │ └── fpn_model.py ├── linknet │ ├── linknet_decoder.py │ └── linknet_model.py ├── manet │ ├── manet_decoder.py │ └── manet_model.py ├── pan │ ├── pan_decoder.py │ └── pan_model.py ├── pspnet │ ├── pspnet_decoder.py │ └── pspnet_model.py ├── unet │ ├── unet_decoder.py │ └── unet_model.py └── unetplusplus │ ├── unetplusplus_decoder.py │ └── unetplusplus_model.py ├── requirements.txt └── utils ├── __init__.py ├── base.py ├── functional.py ├── losses.py ├── meter.py ├── metrics.py ├── modules.py └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/README.md -------------------------------------------------------------------------------- /data/Birds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/data/Birds.py -------------------------------------------------------------------------------- /data/CamVid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/data/CamVid.py -------------------------------------------------------------------------------- /data/CityScapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/data/CityScapes.py -------------------------------------------------------------------------------- /data/LiTS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/data/LiTS.py -------------------------------------------------------------------------------- /data/SkinLesion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/data/SkinLesion.py -------------------------------------------------------------------------------- /dataset/camvid/test/0001TP_008550.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/test/0001TP_008550.png -------------------------------------------------------------------------------- /dataset/camvid/test/0001TP_008580.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/test/0001TP_008580.png -------------------------------------------------------------------------------- /dataset/camvid/test/0001TP_008610.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/test/0001TP_008610.png -------------------------------------------------------------------------------- /dataset/camvid/test/0001TP_008640.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/test/0001TP_008640.png -------------------------------------------------------------------------------- /dataset/camvid/test/0001TP_008670.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/test/0001TP_008670.png -------------------------------------------------------------------------------- /dataset/camvid/test/0001TP_008700.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/test/0001TP_008700.png -------------------------------------------------------------------------------- /dataset/camvid/testannot/0001TP_008550.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/testannot/0001TP_008550.png -------------------------------------------------------------------------------- /dataset/camvid/testannot/0001TP_008580.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/testannot/0001TP_008580.png -------------------------------------------------------------------------------- /dataset/camvid/testannot/0001TP_008610.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/testannot/0001TP_008610.png -------------------------------------------------------------------------------- /dataset/camvid/testannot/0001TP_008640.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/testannot/0001TP_008640.png -------------------------------------------------------------------------------- /dataset/camvid/testannot/0001TP_008670.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/testannot/0001TP_008670.png -------------------------------------------------------------------------------- /dataset/camvid/testannot/0001TP_008700.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/testannot/0001TP_008700.png -------------------------------------------------------------------------------- /dataset/camvid/train/0001TP_006690.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/train/0001TP_006690.png -------------------------------------------------------------------------------- /dataset/camvid/train/0001TP_006720.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/train/0001TP_006720.png -------------------------------------------------------------------------------- /dataset/camvid/train/0001TP_006750.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/train/0001TP_006750.png -------------------------------------------------------------------------------- /dataset/camvid/train/0001TP_006780.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/train/0001TP_006780.png -------------------------------------------------------------------------------- /dataset/camvid/train/0001TP_006810.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/train/0001TP_006810.png -------------------------------------------------------------------------------- /dataset/camvid/train/0001TP_006840.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/train/0001TP_006840.png -------------------------------------------------------------------------------- /dataset/camvid/trainannot/0001TP_006690.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/trainannot/0001TP_006690.png -------------------------------------------------------------------------------- /dataset/camvid/trainannot/0001TP_006720.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/trainannot/0001TP_006720.png -------------------------------------------------------------------------------- /dataset/camvid/trainannot/0001TP_006750.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/trainannot/0001TP_006750.png -------------------------------------------------------------------------------- /dataset/camvid/trainannot/0001TP_006780.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/trainannot/0001TP_006780.png -------------------------------------------------------------------------------- /dataset/camvid/trainannot/0001TP_006810.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/trainannot/0001TP_006810.png -------------------------------------------------------------------------------- /dataset/camvid/trainannot/0001TP_006840.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/trainannot/0001TP_006840.png -------------------------------------------------------------------------------- /dataset/camvid/val/0016E5_07959.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/val/0016E5_07959.png -------------------------------------------------------------------------------- /dataset/camvid/val/0016E5_07961.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/val/0016E5_07961.png -------------------------------------------------------------------------------- /dataset/camvid/val/0016E5_07963.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/val/0016E5_07963.png -------------------------------------------------------------------------------- /dataset/camvid/val/0016E5_07965.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/val/0016E5_07965.png -------------------------------------------------------------------------------- /dataset/camvid/val/0016E5_07967.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/val/0016E5_07967.png -------------------------------------------------------------------------------- /dataset/camvid/val/0016E5_07969.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/val/0016E5_07969.png -------------------------------------------------------------------------------- /dataset/camvid/valannot/0016E5_07959.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/valannot/0016E5_07959.png -------------------------------------------------------------------------------- /dataset/camvid/valannot/0016E5_07961.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/valannot/0016E5_07961.png -------------------------------------------------------------------------------- /dataset/camvid/valannot/0016E5_07963.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/valannot/0016E5_07963.png -------------------------------------------------------------------------------- /dataset/camvid/valannot/0016E5_07965.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/valannot/0016E5_07965.png -------------------------------------------------------------------------------- /dataset/camvid/valannot/0016E5_07967.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/valannot/0016E5_07967.png -------------------------------------------------------------------------------- /dataset/camvid/valannot/0016E5_07969.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/dataset/camvid/valannot/0016E5_07969.png -------------------------------------------------------------------------------- /figures/method.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/figures/method.jpg -------------------------------------------------------------------------------- /figures/overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/figures/overview.jpg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/main.py -------------------------------------------------------------------------------- /models/attonimage/encoder_channelatt_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/attonimage/encoder_channelatt_img.py -------------------------------------------------------------------------------- /models/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/__init__.py -------------------------------------------------------------------------------- /models/base/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/_base.py -------------------------------------------------------------------------------- /models/base/_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/_preprocessing.py -------------------------------------------------------------------------------- /models/base/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/_utils.py -------------------------------------------------------------------------------- /models/base/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/heads.py -------------------------------------------------------------------------------- /models/base/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/initialization.py -------------------------------------------------------------------------------- /models/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/model.py -------------------------------------------------------------------------------- /models/base/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/base/modules.py -------------------------------------------------------------------------------- /models/deeplabv3/deeplabv3_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/deeplabv3/deeplabv3_decoder.py -------------------------------------------------------------------------------- /models/deeplabv3/deeplabv3_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/deeplabv3/deeplabv3_model.py -------------------------------------------------------------------------------- /models/fpn/fpn_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/fpn/fpn_decoder.py -------------------------------------------------------------------------------- /models/fpn/fpn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/fpn/fpn_model.py -------------------------------------------------------------------------------- /models/linknet/linknet_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/linknet/linknet_decoder.py -------------------------------------------------------------------------------- /models/linknet/linknet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/linknet/linknet_model.py -------------------------------------------------------------------------------- /models/manet/manet_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/manet/manet_decoder.py -------------------------------------------------------------------------------- /models/manet/manet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/manet/manet_model.py -------------------------------------------------------------------------------- /models/pan/pan_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/pan/pan_decoder.py -------------------------------------------------------------------------------- /models/pan/pan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/pan/pan_model.py -------------------------------------------------------------------------------- /models/pspnet/pspnet_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/pspnet/pspnet_decoder.py -------------------------------------------------------------------------------- /models/pspnet/pspnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/pspnet/pspnet_model.py -------------------------------------------------------------------------------- /models/unet/unet_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/unet/unet_decoder.py -------------------------------------------------------------------------------- /models/unet/unet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/unet/unet_model.py -------------------------------------------------------------------------------- /models/unetplusplus/unetplusplus_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/unetplusplus/unetplusplus_decoder.py -------------------------------------------------------------------------------- /models/unetplusplus/unetplusplus_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/models/unetplusplus/unetplusplus_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/base.py -------------------------------------------------------------------------------- /utils/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/functional.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/modules.py -------------------------------------------------------------------------------- /utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengtianSang/AFMA/HEAD/utils/train.py --------------------------------------------------------------------------------