MobileNet v3是Google在2019年提出的輕量級卷積神經網絡結構,旨在提高在移動設備上的速度和準確性,廣泛的用于輕量級網絡。
MobileNet v3-Small的網絡結構如下,它的輸入是224x224的3通道彩色圖片。
使用過程如下:
1.創建模型、修改最終分類數量
#1.創建mobilenet_v3_small 模型self.model = torchvision.models.mobilenet_v3_small(pretrained=True)# 2.修改分類數量(二分類)self.model.classifier[3] = nn.Linear(self.model.classifier[3].in_features, num_class) # 修改最后一層# 3.將模型移動到指定的計算設備