code: https://github.com/mahaoxiang822/Boundary-Guided-Context-Aggregation/blob/main/model/BCANet.py importtorchimporttorch.nn.functionalasFfromtorchimportnnfrommodel.dilated_resnetimportget_resnet50_baseline,get_resnet101_baselineclassBCA(nn.Module):def__init__(self,xin_channels,yin_channels,mid_...
(1) Multi-Scale Boundary (MSB) extractor:聚合backbone的不同深度特征图,用于提取图像的边缘信息并预测边缘图; (2) Boundary guided Context Aggregation (BCA) module:利用上述获得的边缘图引导上下文信息聚合,获得鲁棒性更强的特征图; 总结来说,MSB模块进行边缘检测提取边缘图,则BCA模块利用边缘图进行上下文信息完...