Learning a Deep Multi-Scale Feature Ensemble andan Edge-Attention Guidance for Image F(红外图像和可见光图像1)

一、Learning a Deep Multi-Scale Feature Ensemble and
an Edge-Attention Guidance for Image Fusion(基于深度多尺度特征集成的图像融合边缘注意引导)

代码分成了三个部分

(1):Extractor 

def __init__(self):
    super(Extractor, self).__init__()

    # group S
    self.conv_1 = ConvBlock(1, 16, p=1)

    # group A
    self.conv_a1 = ConvBlock(16, 32, p=1)
    self.conv_a2 = ConvBlock(32, 48, p=1)
    self.conv_a3 = ConvBlock(48, 64, p=1)

    # group B
    self.conv_b1 = ConvBlock(16, 32, p=2, d=2)
    self.conv_b2 = ConvBlock(32, 48, p=1)
    self.conv_b3 = ConvBlock(48, 64, p=1)

    # group C
    self.conv_c1 = ConvBlock(16, 32, p=3, d=3)
    self.conv_c2 = ConvBlock(32, 48, p=1)
    self.conv_c3 = ConvBlock(48, 64, p=1)

(2):Constructor

self.conv_1 = ConvBlock(64, 16, p=1)
self.conv_2 = ConvBlock(64, 32, p=1)
self.conv_3 = ConvBlock(64, 16, p=1)
self.conv_4 = ConvBlock(16, 1, p=1)

(3):边缘导向注意机制的过程

self.conv_1 = ConvConv(1, 32, 32)
self.conv_2 = ConvConv(32, 64, 128)
self.conv_3 = ConvConv(128, 64, 32)
self.conv_4 = nn.Conv2d(32, 1, (1, 1))

self.ed = EdgeDetect()

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值