文件名称：Non-local_pytorch:非本地块的实现
文件大小：1.52MB
文件格式：ZIP
更新时间：2024-05-31 18:50:31
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()