非本地_火炬 。 陈述 您可以在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
2021-12-13 15:08:43 1.52MB pytorch non-local non-local-block nonlocal
1