We propose Uncertainty Augmented Context Attention network (UACANet) for polyp segmentation which consider a uncertain area of the saliency map. We construct a modified version of U-Net shape network with additional encoder and decoder and compute a saliency map in each bottom-up stream prediction ...
UACANet: Uncertainty Augmented Context Attention for Polyp Segmentationdoi:10.1145/3474085.3475375Taehun KimHyemin LeeDaijin KimACM
# Single GPU CUDA_VISIBLE_DEVICES=0 python run/Train.py --config configs/UACANet-L.yaml --verbose --debug # Multi GPU (e.g., 0 and 1) CUDA_VISIBLE_DEVICES=0,1 python -m torchrun --nproc_per_node 2 run/Train.py --config configs/UACANet-L.yaml --verbose --debug ...