shufflenetv2-1.0x-imagenet-torch#

Mobile image classifier that works efficiently on phones with modest computing resources.

Details

  • Model name: shufflenetv2-1.0x-imagenet-torch

  • Model source: https://pytorch.org/vision/main/models.html

  • Model author: Ningning Ma, et al.

  • Model license: BSD 3-Clause

  • Model size: 8.79 MB

  • Exposes embeddings? yes

  • Tags: classification, embeddings, logits, imagenet, torch, shufflenet, official

Requirements

  • Packages: torch, torchvision

  • CPU support

    • yes

  • GPU support

    • yes

Example usage

 1import fiftyone as fo
 2import fiftyone.zoo as foz
 3
 4dataset = foz.load_zoo_dataset(
 5    "imagenet-sample",
 6    dataset_name=fo.get_default_dataset_name(),
 7    max_samples=50,
 8    shuffle=True,
 9)
10
11model = foz.load_zoo_model("shufflenetv2-1.0x-imagenet-torch")
12
13dataset.apply_model(model, label_field="predictions")
14
15session = fo.launch_app(dataset)