预训练 ResNet18 模型注入方法
在预训练的 ResNet18 模型中注入自定义模块或修改现有结构时，需确保不破坏原有特征提取能力。以下是常见注入方法：
修改特定层结构通过替换或扩展特定层（如卷积层、全连接层）实现功能注入。例如，在最后一层前插入自定义模块：
import torch.nn as nn from torchvision.models import resnet18 model = resnet18(pretrained=True) model.fc = nn.Sequential( nn.Linear(512, 256), nn.ReLU(), nn.Linear(256, 10) # 修改输出维度 )插入注意力机制在中间层注入注意力模块（如SEBlock）：
class SEBlock(nn.Module): def __init__(self, channels, reduction=16): super().__init__() self.squeeze = nn.AdaptiveAvgPool2d(1) self.excitation = nn.Sequential( nn.Linear(channels, channels // reduction), nn.ReLU(), nn.Linear(channels // reduction, channels), nn.Sigmoid() ) model.layer2.add_module("se_block", SEBlock(128))特征融合注入通过跳跃连接将不同层特征融合：
class FusionModule(nn.Module): def __init__(self, in_channels): super().__init__() self.conv = nn.Conv2d(in_channels*2, in_channels, 1) def forward(self, x1, x2): x = torch.cat([x1, x2], dim=1) return self.conv(x) fusion = FusionModule(64) model.layer1.register_forward_hook(lambda m, inp, out: fusion(out, inp[0]))注入后的微调策略
分层学习率设置对注入模块使用更高学习率，预训练层采用较低学习率：
optimizer = torch.optim.SGD([ {'params': model.parameters(), 'lr': 0.001}, {'params': fusion.parameters(), 'lr': 0.01} ], momentum=0.9)渐进式解冻先冻结所有层仅训练注入模块，逐步解冻深层：
for param in model.parameters(): param.requires_grad = False fusion.train() # 后续按epoch逐步解冻layer4、layer3等验证注入效果
特征可视化使用Grad-CAM等工具比较注入前后特征响应：
from torchcam.methods import GradCAM cam_extractor = GradCAM(model, 'layer4') activations = cam_extractor(input_tensor)性能指标对比记录注入前后的验证集准确率、推理速度等关键指标，确保注入不会显著降低模型效率。典型评估流程：
with torch.no_grad(): for data in val_loader: outputs = model(data) # 计算accuracy, F1-score等rogbbs.asus.com.cn/topics/101/feeds/63030
rogbbs.asus.com.cn/topics/101/feeds/63031
rogbbs.asus.com.cn/topics/101/feeds/63032
rogbbs.asus.com.cn/topics/101/feeds/63033
rogbbs.asus.com.cn/topics/101/feeds/63034
rogbbs.asus.com.cn/topics/101/feeds/63035
rogbbs.asus.com.cn/topics/101/feeds/63036
rogbbs.asus.com.cn/topics/101/feeds/63037
rogbbs.asus.com.cn/topics/101/feeds/63038
rogbbs.asus.com.cn/topics/101/feeds/63039
rogbbs.asus.com.cn/topics/101/feeds/63040
rogbbs.asus.com.cn/topics/101/feeds/63041
rogbbs.asus.com.cn/topics/101/feeds/63044
rogbbs.asus.com.cn/topics/101/feeds/63045
rogbbs.asus.com.cn/topics/101/feeds/63047
rogbbs.asus.com.cn/topics/101/feeds/63048
rogbbs.asus.com.cn/topics/101/feeds/63049
rogbbs.asus.com.cn/topics/101/feeds/63050
rogbbs.asus.com.cn/topics/101/feeds/63046
rogbbs.asus.com.cn/topics/101/feeds/63052
rogbbs.asus.com.cn/topics/101/feeds/63053
rogbbs.asus.com.cn/topics/101/feeds/63043
rogbbs.asus.com.cn/topics/101/feeds/63055
rogbbs.asus.com.cn/topics/101/feeds/63056
rogbbs.asus.com.cn/topics/101/feeds/63057
rogbbs.asus.com.cn/topics/101/feeds/63058
rogbbs.asus.com.cn/topics/101/feeds/63059
rogbbs.asus.com.cn/topics/101/feeds/63062
rogbbs.asus.com.cn/topics/101/feeds/63060
rogbbs.asus.com.cn/topics/101/feeds/63051
rogbbs.asus.com.cn/topics/101/feeds/63063
rogbbs.asus.com.cn/topics/101/feeds/63064
rogbbs.asus.com.cn/topics/101/feeds/63067
rogbbs.asus.com.cn/topics/101/feeds/63066
rogbbs.asus.com.cn/topics/101/feeds/63070
rogbbs.asus.com.cn/topics/101/feeds/63071
rogbbs.asus.com.cn/topics/101/feeds/63072
rogbbs.asus.com.cn/topics/101/feeds/63073
rogbbs.asus.com.cn/topics/101/feeds/63069
rogbbs.asus.com.cn/topics/101/feeds/63074
rogbbs.asus.com.cn/topics/101/feeds/63075
rogbbs.asus.com.cn/topics/101/feeds/63076
rogbbs.asus.com.cn/topics/101/feeds/63078
rogbbs.asus.com.cn/topics/101/feeds/63079
rogbbs.asus.com.cn/topics/101/feeds/63080
rogbbs.asus.com.cn/topics/101/feeds/63082
rogbbs.asus.com.cn/topics/101/feeds/63084
rogbbs.asus.com.cn/topics/101/feeds/63085
rogbbs.asus.com.cn/topics/101/feeds/63086
rogbbs.asus.com.cn/topics/101/feeds/63088
rogbbs.asus.com.cn/topics/101/feeds/63087
rogbbs.asus.com.cn/topics/101/feeds/63089
rogbbs.asus.com.cn/topics/101/feeds/63090
rogbbs.asus.com.cn/topics/101/feeds/63091
rogbbs.asus.com.cn/topics/101/feeds/63092
rogbbs.asus.com.cn/topics/101/feeds/63093
rogbbs.asus.com.cn/topics/101/feeds/63094
rogbbs.asus.com.cn/topics/101/feeds/63095
rogbbs.asus.com.cn/topics/101/feeds/63096
rogbbs.asus.com.cn/topics/101/feeds/63097
rogbbs.asus.com.cn/topics/101/feeds/63098
rogbbs.asus.com.cn/topics/101/feeds/63099
rogbbs.asus.com.cn/topics/101/feeds/63100
rogbbs.asus.com.cn/topics/101/feeds/63101
rogbbs.asus.com.cn/topics/101/feeds/63102
rogbbs.asus.com.cn/topics/101/feeds/63103
rogbbs.asus.com.cn/topics/101/feeds/63104
rogbbs.asus.com.cn/topics/101/feeds/63107
rogbbs.asus.com.cn/topics/101/feeds/63105
rogbbs.asus.com.cn/topics/101/feeds/63106
rogbbs.asus.com.cn/topics/101/feeds/63108
rogbbs.asus.com.cn/topics/101/feeds/63109
rogbbs.asus.com.cn/topics/101/feeds/63110
rogbbs.asus.com.cn/topics/101/feeds/63111
rogbbs.asus.com.cn/topics/101/feeds/63112
rogbbs.asus.com.cn/topics/101/feeds/63113
rogbbs.asus.com.cn/topics/101/feeds/63114
rogbbs.asus.com.cn/topics/101/feeds/63115
rogbbs.asus.com.cn/topics/101/feeds/63116
rogbbs.asus.com.cn/topics/101/feeds/63117
rogbbs.asus.com.cn/topics/101/feeds/63118
rogbbs.asus.com.cn/topics/101/feeds/63119
rogbbs.asus.com.cn/topics/101/feeds/63121
rogbbs.asus.com.cn/topics/101/feeds/63122
rogbbs.asus.com.cn/topics/101/feeds/63123
rogbbs.asus.com.cn/topics/101/feeds/63124
rogbbs.asus.com.cn/topics/101/feeds/63125
rogbbs.asus.com.cn/topics/101/feeds/63126
rogbbs.asus.com.cn/topics/101/feeds/63127
rogbbs.asus.com.cn/topics/101/feeds/63128
rogbbs.asus.com.cn/topics/101/feeds/63129
rogbbs.asus.com.cn/topics/101/feeds/63130
rogbbs.asus.com.cn/topics/101/feeds/63131
rogbbs.asus.com.cn/topics/101/feeds/63132
rogbbs.asus.com.cn/topics/101/feeds/63133
rogbbs.asus.com.cn/topics/101/feeds/63134
rogbbs.asus.com.cn/topics/101/feeds/63135
rogbbs.asus.com.cn/topics/101/feeds/63136
rogbbs.asus.com.cn/topics/101/feeds/63137
rogbbs.asus.com.cn/topics/101/feeds/63138
rogbbs.asus.com.cn/topics/101/feeds/63139
rogbbs.asus.com.cn/topics/101/feeds/63140
rogbbs.asus.com.cn/topics/101/feeds/63142
rogbbs.asus.com.cn/topics/101/feeds/63143
rogbbs.asus.com.cn/topics/101/feeds/63144
rogbbs.asus.com.cn/topics/101/feeds/63146
rogbbs.asus.com.cn/topics/101/feeds/63145
rogbbs.asus.com.cn/topics/101/feeds/63120
rogbbs.asus.com.cn/topics/101/feeds/63148
rogbbs.asus.com.cn/topics/101/feeds/63150
rogbbs.asus.com.cn/topics/101/feeds/63149
rogbbs.asus.com.cn/topics/101/feeds/63151
rogbbs.asus.com.cn/topics/101/feeds/63152
rogbbs.asus.com.cn/topics/101/feeds/63154
rogbbs.asus.com.cn/topics/101/feeds/63153
rogbbs.asus.com.cn/topics/101/feeds/63141
rogbbs.asus.com.cn/topics/101/feeds/63155
rogbbs.asus.com.cn/topics/101/feeds/63156
rogbbs.asus.com.cn/topics/101/feeds/63157
rogbbs.asus.com.cn/topics/101/feeds/63158
rogbbs.asus.com.cn/topics/101/feeds/63159
rogbbs.asus.com.cn/topics/101/feeds/63160
rogbbs.asus.com.cn/topics/101/feeds/63161
rogbbs.asus.com.cn/topics/101/feeds/63162
rogbbs.asus.com.cn/topics/101/feeds/63163
rogbbs.asus.com.cn/topics/101/feeds/63164
rogbbs.asus.com.cn/topics/101/feeds/63165
rogbbs.asus.com.cn/topics/101/feeds/63167
rogbbs.asus.com.cn/topics/101/feeds/63168
rogbbs.asus.com.cn/topics/101/feeds/63190
rogbbs.asus.com.cn/topics/101/feeds/63166
rogbbs.asus.com.cn/topics/101/feeds/63184
rogbbs.asus.com.cn/topics/101/feeds/63177
rogbbs.asus.com.cn/topics/101/feeds/63178
rogbbs.asus.com.cn/topics/101/feeds/63174
rogbbs.asus.com.cn/topics/101/feeds/63169
rogbbs.asus.com.cn/topics/101/feeds/63173
rogbbs.asus.com.cn/topics/101/feeds/63187
rogbbs.asus.com.cn/topics/101/feeds/63179
rogbbs.asus.com.cn/topics/101/feeds/63180
rogbbs.asus.com.cn/topics/101/feeds/63185
rogbbs.asus.com.cn/topics/101/feeds/63175
rogbbs.asus.com.cn/topics/101/feeds/63172
rogbbs.asus.com.cn/topics/101/feeds/63181
rogbbs.asus.com.cn/topics/101/feeds/63186
rogbbs.asus.com.cn/topics/101/feeds/63182
rogbbs.asus.com.cn/topics/101/feeds/63183
rogbbs.asus.com.cn/topics/101/feeds/63188
rogbbs.asus.com.cn/topics/101/feeds/63176
rogbbs.asus.com.cn/topics/101/feeds/63170
rogbbs.asus.com.cn/topics/101/feeds/63189
rogbbs.asus.com.cn/topics/101/feeds/63191
rogbbs.asus.com.cn/topics/101/feeds/63192
rogbbs.asus.com.cn/topics/101/feeds/63193
rogbbs.asus.com.cn/topics/101/feeds/63194
rogbbs.asus.com.cn/topics/101/feeds/63195
rogbbs.asus.com.cn/topics/101/feeds/63196
rogbbs.asus.com.cn/topics/101/feeds/63197
rogbbs.asus.com.cn/topics/101/feeds/63199
rogbbs.asus.com.cn/topics/101/feeds/63200
rogbbs.asus.com.cn/topics/101/feeds/63201
rogbbs.asus.com.cn/topics/101/feeds/63203
rogbbs.asus.com.cn/topics/101/feeds/63204
rogbbs.asus.com.cn/topics/101/feeds/63205
rogbbs.asus.com.cn/topics/101/feeds/63206
rogbbs.asus.com.cn/topics/101/feeds/63208
rogbbs.asus.com.cn/topics/101/feeds/63207
rogbbs.asus.com.cn/topics/101/feeds/63209
rogbbs.asus.com.cn/topics/101/feeds/63210
rogbbs.asus.com.cn/topics/101/feeds/63211
rogbbs.asus.com.cn/topics/101/feeds/63212
rogbbs.asus.com.cn/topics/101/feeds/63213
rogbbs.asus.com.cn/topics/101/feeds/63215
rogbbs.asus.com.cn/topics/101/feeds/63214
rogbbs.asus.com.cn/topics/101/feeds/63216
rogbbs.asus.com.cn/topics/101/feeds/63217
rogbbs.asus.com.cn/topics/101/feeds/63218
rogbbs.asus.com.cn/topics/101/feeds/63220
rogbbs.asus.com.cn/topics/101/feeds/63219
rogbbs.asus.com.cn/topics/101/feeds/63221
rogbbs.asus.com.cn/topics/101/feeds/63224
rogbbs.asus.com.cn/topics/101/feeds/63222
rogbbs.asus.com.cn/topics/101/feeds/63223
rogbbs.asus.com.cn/topics/101/feeds/63225
rogbbs.asus.com.cn/topics/101/feeds/63226
rogbbs.asus.com.cn/topics/101/feeds/63227
rogbbs.asus.com.cn/topics/101/feeds/63228
rogbbs.asus.com.cn/topics/101/feeds/63229
rogbbs.asus.com.cn/topics/101/feeds/63231
rogbbs.asus.com.cn/topics/101/feeds/63230
rogbbs.asus.com.cn/topics/101/feeds/63233
rogbbs.asus.com.cn/topics/101/feeds/63234
rogbbs.asus.com.cn/topics/101/feeds/63232
rogbbs.asus.com.cn/topics/101/feeds/63235
rogbbs.asus.com.cn/topics/101/feeds/63236
rogbbs.asus.com.cn/topics/101/feeds/63237
rogbbs.asus.com.cn/topics/101/feeds/63238
rogbbs.asus.com.cn/topics/101/feeds/63239
rogbbs.asus.com.cn/topics/101/feeds/63240
rogbbs.asus.com.cn/topics/101/feeds/63241
rogbbs.asus.com.cn/topics/101/feeds/63242
rogbbs.asus.com.cn/topics/101/feeds/63247
rogbbs.asus.com.cn/topics/101/feeds/63248
rogbbs.asus.com.cn/topics/101/feeds/63250
rogbbs.asus.com.cn/topics/101/feeds/63245
rogbbs.asus.com.cn/topics/101/feeds/63244
rogbbs.asus.com.cn/topics/101/feeds/63253
rogbbs.asus.com.cn/topics/101/feeds/63254
rogbbs.asus.com.cn/topics/101/feeds/63246
rogbbs.asus.com.cn/topics/101/feeds/63256
rogbbs.asus.com.cn/topics/101/feeds/63255
rogbbs.asus.com.cn/topics/101/feeds/63257
rogbbs.asus.com.cn/topics/101/feeds/63258
rogbbs.asus.com.cn/topics/101/feeds/63259
rogbbs.asus.com.cn/topics/101/feeds/63260
rogbbs.asus.com.cn/topics/101/feeds/63261
rogbbs.asus.com.cn/topics/101/feeds/63262
rogbbs.asus.com.cn/topics/101/feeds/63263
rogbbs.asus.com.cn/topics/101/feeds/63264
rogbbs.asus.com.cn/topics/101/feeds/63265
rogbbs.asus.com.cn/topics/101/feeds/63266
)