CCNet-Pure-Pytorch 用于纯Pytorch中语义分割的Criss-Cross Attention(2d&3d),具有更快,更精确的实现方式。 更新 **** 2021/03:纯pytorch实现3D CCNET模块的三种被释放 。 您可以在和检查其正确性 介绍 我非正式地重新实现了纯Pytorch中的以便在不同版本和环境下实现更好的兼容性。 以前的许多开源项目都在Pytorch上使用了Cuda扩展,因此存在兼容性和精度损失的问题。 此外,当我们设置cudnn.benchmark = True时,Pytorch可能无法优化和加速Cuda扩展。 为了解决这些问题,我基于的张量变换在CC.py中设计了一个Criss-Cross Attention操作,该操作并行执行,并且在向前结果和向后渐变中显示出更快的速度和更精确的效果。 我的运作和表现 不需要CUDA扩展。 以前的“ Cr
2021-11-18 14:16:15 3.96MB tensorflow pytorch attention ccnet
1
PyTorch上的语义分割 该项目旨在为使用PyTorch的语义细分模型提供简洁,易用,可修改的参考实现。 安装 # semantic-segmentation-pytorch dependencies pip install ninja tqdm # follow PyTorch installation in https://pytorch.org/get-started/locally/ conda install pytorch torchvision -c pytorch # install PyTorch Segmentation git clone https://github.com/Tramac/awesome-semantic-segmentation-pytorch.git # the following will install the lib with symb
2021-10-11 16:39:42 653KB pytorch semantic-segmentation Python
1
注意力机制网络
2021-04-24 22:01:27 605KB 深度搜索
1
CCNet:语义细分的跨界关注
2021-02-26 19:02:46 833KB 研究论文
1