UACANet: Uncertainty Augmented Context Attention for Polyp Segmentation代码
看一下整体的结构:首先按照惯例我们测试一下模型的参数和计算量:defmain():model=UACANet(channels=256,output_stride=16,pretrained=True)#(传入参数)model.eval()rgb=torch.randn(1,3,352,352)summary(model,input_size=[(3,352,352)],device='cpu')