为简便起见,此处使用来自 torchvision
的 mobilenet_v2。
from torchvision.models.mobilenet import mobilenet_v2
model = mobilenet_v2(pretrained=True)
为简便起见,此处使用来自 torchvision
的 mobilenet_v2。
from torchvision.models.mobilenet import mobilenet_v2
model = mobilenet_v2(pretrained=True)