Non-local_pytorch:非本地块的实现-源码

非本地_火炬 。 陈述 您可以在lib /中找到不同种类的非本地块。 您可以按照以下显示的“运行步骤”形象化Non_local Attention Map。 该代码在MNIST数据集上进行了测试。 您可以在lib / network.py中选择非本地块的类型。 如果我的代码有问题，请与我联系，谢谢！ 环境 python 3.7.7 pytorch 1.4.0（> = 1.2.0，1.8.1也可以） opencv 4.2.0.34（其他人也可以） 可视化 在第一个非本地层中。 在第二个非本地层中。 运行步骤 在lib / network.py中选择非本地块的类型。 from lib.non_local_concatenation import NONLocalBlock2D from lib.non_local_gaussian import NONLocalBlock2D f