簡単にするために、ここでは 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)