Shortcuts

Source code for torchexpo.vision.image_classification.mobilenet

import torchvision
from torchexpo.modules import ImageClassificationModule


[docs]def mobilenet_v2(): """MobileNet V2 Model pre-trained on ImageNet""" model = torchvision.models.mobilenet_v2(pretrained=True) obj = ImageClassificationModule(model, "MobileNet_V2", model_example="default") return obj
Read the Docs v: latest
Versions
latest
Downloads
pdf
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.