基于resnet50_cam的改进

 主要目标是进行图像分类任务,并且通过 ASPP 模块提高网络的感受野,CAM 类则用于生成类激活映射,帮助理解模型对输入图像的分类依据。

ASPP类 Atrous spatial pyramid pooling

用于分类任务,其中包含带有不同扩张率的卷积层。

Net类

对输入图像和标签进行处理,通过骨干网络和线性变换,最终通过 ASPP 进行分类。

初始化方法

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()

        self.resnet50 = ···
        self.stage1 = ···
        self.stage2 = ···
        self.stage3 = ···
        self.stage4 =···

 引入线性层和ASPP模块

''' 线性层 '''
        self.label_enc = nn.Linear(in_features, out_features)
        ''' ASPP 模块,用于分类任务,其中包含带有不同扩张率的卷积层'''
        self.classifier = nn.Sequential(
            nn.Dropout(a),
            _ASPP(in_ch, out_ch, rates=astrous_rates)
        )

主干和新增

        self.backbone
        self.newly_added

 forward

首先对输入的标签进行线性变换,并

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值