MobileNetV2 Model — model_mobilenet_v2 • torchvision

Constructs a MobileNetV2 architecture from MobileNetV2: Inverted Residuals and Linear Bottlenecks.

model_mobilenet_v2(pretrained = FALSE, progress = TRUE, ...)

Arguments

pretrained: (bool): If TRUE, returns a model pre-trained on ImageNet.
progress: (bool): If TRUE, displays a progress bar of the download to stderr.
...: Other parameters passed to the model implementation.

See also

Other classification_model: model_alexnet(), model_convnext, model_efficientnet, model_efficientnet_v2, model_inception_v3(), model_maxvit(), model_mobilenet_v3, model_resnet, model_vgg, model_vit