Inefficient layout weight

本文介绍了一种通过将布局宽度设置为0dp而非match_parent的方法来提高应用性能的技术。这种做法能够有效地减少不必要的视图测量,从而提升应用程序的整体运行效率。
Use a 'layout_width' of '0dp' instead of 'match_parent' for better performance
mport torch import torch.nn as nn class CBAM(nn.Module): def __init__(self, channels, reduction=16): super(CBAM, self).__init__() self.avg_pool = nn.AdaptiveAvgPool2d(1) self.max_pool = nn.AdaptiveMaxPool2d(1) self.fc = nn.Sequential( nn.Linear(channels, channels // reduction), nn.ReLU(inplace=True), nn.Linear(channels // reduction, channels) ) self.sigmoid = nn.Sigmoid() def forward(self, x): avg_out = self.fc(self.avg_pool(x).view(x.size(0), -1)) max_out = self.fc(self.max_pool(x).view(x.size(0), -1)) out = avg_out + max_out channel_att = self.sigmoid(out).unsqueeze(2).unsqueeze(3) x = x * channel_att return x class ResNet50_CBAM(nn.Module): def __init__(self, num_classes=1000): super(ResNet50_CBAM, self).__init__() self.base = torchvision.models.resnet50(pretrained=True) self.cbam1 = CBAM(256) self.cbam2 = CBAM(512) self.cbam3 = CBAM(1024) self.cbam4 = CBAM(2048) self.fc = ArcFace(2048, num_classes) def forward(self, x): x = self.base.conv1(x) x = self.base.bn1(x) x = self.base.relu(x) x = self.base.maxpool(x) x = self.base.layer1(x) x = self.cbam1(x) x = self.base.layer2(x) x = self.cbam2(x) x = self.base.layer3(x) x = self.cbam3(x) x = self.base.layer4(x) x = self.cbam4(x) x = self.base.avgpool(x) x = torch.flatten(x, 1) x = self.fc(x) return x torch import torch.nn as nn import torch.nn.functional as F from torch.nn import Parameter class ArcFace(nn.Module): def __init__(self, in_features, out_features, s=30.0, m=0.50): super(ArcFace, self).__init__() self.in_features = in_features self.out_features = out_features self.s = s self.m = m self.weight = Parameter(torch.FloatTensor(out_features, in_features)) nn.init.xavier_uniform_(self.weight) def forward(self, input, label=None): cosine = F.linear(F.normalize(input), F.normalize(self.weight)) if label is None: return cosine * self.s phi = torch.acos(torch.clamp(cosine, -1.0 + 1e-7, 1.0 - 1e-7)) one_hot = torch.zeros(cosine.size(), device=input.device) one_hot.scatter_(1, label.view(-1, 1), 1) output = (one_hot * (phi + self.m) + (1.0 - one_hot) * phi).cos() output *= self.s return output我想要让这段代码在不该的情况下有输出结果
07-19
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值