7.5 使用MobileNet v3进行图像的区分
MobileNet v3是Google在2019年提出的轻量级卷积神经网络结构，旨在提高在移动设备上的速度和准确性，广泛的用于轻量级网络。MobileNet v3-Small的网络结构如下，它的输入是224x224的3通道彩色图片。
https://i-blog.csdnimg.cn/direct/65bad53ce56246bf900573e337f372e9.png
使用过程如下：
1.创建模子、修改最终分类数量
#1.创建mobilenet_v3_small 模型
self.model = torchvision.models.mobilenet_v3_small(pretrained=True)
# 2.修改分类数量（二分类）
self.model.classifier = nn.Linear(self.model.classifier.in_features, num_class)# 修改最后一层
# 3.将模型移动到指定的计算设备
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]