UACANet: Uncertainty Augmented Context Attention for Polyp Segmentation代码补充
上一篇看了文章创新点的代码,现在看一下train文件等其余的文件。看主函数:importosimporttorchimportargparseimporttqdmimportsysimportcv2importtorch.nnasnnimporttorch.distributedasdistfromtorch.optimimportAdam,SGDfromtorch.cuda.ampimportG