昇思25天学习打卡营第24天|基于MobileNetv2的垃圾分类
创始人
2024-11-14 05:08:26

用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:执行整个模型的正向传播。

相关内容

热门资讯

裸辞做“一人公司”,我后悔了 去年这个时候,一位以色列程序员正在东南亚旅行。他顺手把一个在脑子里转了很久的想法做成了产品,一个让任...
南京建成国内首个Pre-6G试... 4月21日,2026全球6G技术与产业生态大会在南京开幕。全息互动技术展台前,一名远在北京的工作人员...
超梵求职受邀参加“2025抖音... 超梵求职受邀参加“2025抖音巨量引擎成人教育行业生态大会”,探讨分享优质内容传播,服务万千学员。 ...
摩托罗拉Razr 2026(R... IT之家 4 月 22 日消息,摩托罗拉宣布新一代 Razr 折叠手机将于 4 月 29 日在美国发...
库克卸任,特纳斯领航:苹果新纪... 苹果首席执行官蒂姆·库克将卸任,硬件工程主管约翰·特纳斯将接任,苹果公司今天宣布此事。 库克将在夏季...