Skip to content

πŸš€ Lightning-fast computer vision models. Fine-tune SOTA models with just a few lines of code. Ready for cloud ☁️ and edge πŸ“± deployment.

License

Notifications You must be signed in to change notification settings

FocoosAI/focoos

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

FocoosAI

Tests Colab Documentation

Welcome to Focoos AI

Focoos AI provides an advanced development platform designed to empower developers and businesses with efficient, customizable computer vision solutions. Whether you're working with data from cloud infrastructures or deploying on edge devices, Focoos AI enables you to select, fine-tune, and deploy state-of-the-art models optimized for your unique needs.

Overview

The Focoos library is your gateway to easily access cutting-edge computer vision models and development tools. With just a few lines of code, you can fine tune pre-trained models tailored to your specific needs.

Whether you're working in the cloud or on edge devices, the Focoos library seamlessly integrates into your workflow, accelerating development and simplifying the implementation of computer vision solutions.

Key Features πŸ”‘

  1. Frugal Pretrained Models 🌿 Get started quickly by selecting one of our efficient, pre-trained models that best suits your data and application needs. Focoos Model Registry give access to 11 pretrained models of different size from different families: RTDetr, Maskformer, BisenetFormer

  2. Fine Tune Your Model ✨ Adapt the model to your specific use case by customize its config and training it on your own dataset.

  3. Optimized Inference πŸ–₯️ Export Models and run inference efficiently, Leverage hardware acceleration through Torchscript, TensorRT and ONNX for maximum performance.

  4. FocoosHub Integration πŸ”„ Seamlessly integrate with Focoos Cloud to access your models and data, you can also run cloud inference on managed models.

Quickstart πŸš€

Ready to dive in? Get started with the setup in just a few simple steps!

Installation

Install the Focoos library (for more options, see setup)

uv pip install 'focoos @ git+https://github.com/FocoosAI/focoos.git'

Inference

from focoos import ModelManager


im = "https://public.focoos.ai/samples/motogp.jpg" # can be local/remote path, np.array, PIL image

model = ModelManager.get("fai-detr-l-obj365") # any models from ModelRegistry, FocoosHub or local folder

detections = model.infer(im,annotate=true)

Training

from focoos import ModelManager,DatasetSplitType, DatasetLayout, RuntimeType, TrainerArgs, Task
from focoos.data import get_default_by_task
from focoos.data import AutoDataset



ds_name = "my_dataset.zip"
task = Task.DETECTION
layout = DatasetLayout.ROBOFLOW_COCO

auto_dataset = AutoDataset(dataset_name=ds_name, task=task, layout=layout)

train_augs, val_augs = get_default_by_task(task, 640, advanced=False)
train_dataset = auto_dataset.get_split(augs=train_augs.get_augmentations(), split=DatasetSplitType.TRAIN)
valid_dataset = auto_dataset.get_split(augs=val_augs.get_augmentations(), split=DatasetSplitType.VAL)


model = ModelManager.get("fai-detr-l-obj365")

args = TrainerArgs(
    run_name=f"{model.name}_{train_dataset.name}",
    batch_size=16,
    max_iters=50,
    eval_period=50,
    learning_rate=0.0008,
    sync_to_hub=False,  # use this to sync model info, weights and metrics on the hub
)


model.train(args, train_dataset, valid_dataset)

# if you want to do only evaluation

model.eval(args, valid_dataset)

Export, Benchmark and optimized Inference

from focoos import RuntimeType, ModelManager

model = ModelManager.get("fai-detr-l-obj365")
infer_model = model.export(runtime_type=RuntimeType.TORCHSCRIPT_32)
infer_model.benchmark()

Our Models 🧠

Focoos AI offers the best models in object detection, semantic and instance segmentation, and more is coming soon.

Using Focoos AI helps you save both time and money while delivering high-performance AI models πŸ’ͺ:

  • 10x Faster ⏳: Our models are able to process images up to ten times faster than traditional methods.
  • 4x Cheaper πŸ’°: Our models require up to 4x less computational power, letting you save on hardware or cloud bill while ensuring high-quality results.
  • Tons of CO2 saved annually per model 🌱: Our models are energy-efficient, helping you reduce your carbon footprint by using less powerful hardware with respect to mainstream models.

See the list of our models in the models section.


Start now!

By choosing Focoos AI, you can save time, reduce costs, and achieve superior model performance, all while ensuring the privacy and efficiency of your deployments. Reach out to us to ask for your API key for free and power your computer vision projects. πŸš€

❀️ Contributors

Contributors

About

πŸš€ Lightning-fast computer vision models. Fine-tune SOTA models with just a few lines of code. Ready for cloud ☁️ and edge πŸ“± deployment.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 6