microsoft/Florence-2-base#

From Plugin

Note

This is a remotely-sourced model from the florence2 plugin, maintained by the community. It is not part of FiftyOne core and may have special installation requirements. Please review the plugin documentation and license before use.

Florence-2 is a vision foundation model with a unified, prompt-based representation for a variety of computer vision and vision-language tasks (https://arxiv.org/abs/2311.06242)..

Details

Requirements

  • Packages: huggingface-hub, transformers, torch, torchvision, einops, timm, accelerate

  • CPU support

    • yes

  • GPU support

    • yes

Example usage

 1import fiftyone as fo
 2import fiftyone.zoo as foz
 3
 4foz.register_zoo_model_source("https://github.com/harpreetsahota204/florence2")
 5
 6dataset = foz.load_zoo_dataset(
 7    "coco-2017",
 8    split="validation",
 9    dataset_name=fo.get_default_dataset_name(),
10    max_samples=50,
11    shuffle=True,
12)
13
14model = foz.load_zoo_model("microsoft/Florence-2-base")
15
16dataset.apply_model(model, label_field="predictions")
17
18session = fo.launch_app(dataset)