昇思25天学习打卡营第24天|基于MobileNetv2的垃圾分类

用MindSpore搭MobileNetV2这个网络的时候,要继承mindspore.nn.Cell这个类哦。这个网络里的激活函数用的是ReLU6,池化部分用的是全局平均池化。

下面这段代码是从官方文档来的,它定义了一个MobileNetV2模型,里面有个轻量级的主干网(MobileNetV2Backbone)和一个分类头(MobileNetV2Head)。这个模型用了倒置残差块和深度可分离卷积来减轻计算负担,特别适合手机这种设备做图片分类。而且它还给用户提供了很多自定义和扩展的机会。

MobileNetV2类:
作用:整合主干网络和分类头,构建完整的MobileNetV2模型。
参数:
num_classes:分类的类别数。
width_mult:通道数的倍乘因子。
has_dropout:是否使用Dropout层。
inverted_residual_setting:倒置残差块的配置列表。
round_nearest:通道数取整的基数。
input_channel:输入层的通道数。
last_channel:最后一层的通道数。
方法:
construct:执行整个模型的正向传播。

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值