【文件属性】:
文件名称:Non-local_pytorch:非本地块的实现
文件大小:1.52MB
文件格式:ZIP
更新时间:2021-05-12 01:03:51
pytorch non-local non-local-block nonlocal nl-map
非本地_火炬
。
陈述
您可以在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
【文件预览】:
Non-local_pytorch-master
----Non-Local_pytorch_0.3.1()
--------demo_MNIST.py(2KB)
--------lib()
----demo_MNIST_train.py(2KB)
----LICENSE(11KB)
----README.md(4KB)
----nl_map_vis()
--------nl_map_1.npy(38KB)
--------nl_map_2.npy(2KB)
--------nl_map_1()
--------nl_map_2()
--------nl_map_vis.py(2KB)
--------sample.png(254B)
----lib()
--------non_local_dot_product.py(5KB)
--------network.py(2KB)
--------non_local_gaussian.py(5KB)
--------non_local_embedded_gaussian.py(5KB)
--------non_local_concatenation.py(5KB)
----weights()
--------net.pth(1.54MB)
----demo_MNIST_AMP_train_with_single_gpu.py(3KB)
----nl_map_save.py(1KB)
----Non-Local_pytorch_0.4.1_to_1.1.0()
--------demo_MNIST.py(2KB)
--------lib()