Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

hustvl
/
yolos-small

Object Detection
Transformers
PyTorch
Safetensors
yolos
vision
Model card Files Files and versions
xet
Community
4

Instructions to use hustvl/yolos-small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use hustvl/yolos-small with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("object-detection", model="hustvl/yolos-small")
    # Load model directly
    from transformers import AutoImageProcessor, AutoModelForObjectDetection
    
    processor = AutoImageProcessor.from_pretrained("hustvl/yolos-small")
    model = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-small")
  • Notebooks
  • Google Colab
  • Kaggle
yolos-small
246 MB
Ctrl+K
Ctrl+K
  • 2 contributors
History: 12 commits
nielsr's picture
nielsr HF Staff
Update preprocessor_config.json
3d8f713 verified about 2 years ago
  • .gitattributes
    1.23 kB
    Adding `safetensors` variant of this model (#2) about 2 years ago
  • README.md
    4.17 kB
    Add vision tag almost 4 years ago
  • config.json
    4.13 kB
    add model about 4 years ago
  • model.safetensors
    123 MB
    xet
    Adding `safetensors` variant of this model (#2) about 2 years ago
  • preprocessor_config.json
    292 Bytes
    Update preprocessor_config.json about 2 years ago
  • pytorch_model.bin

    Detected Pickle imports (3)

    • "collections.OrderedDict",
    • "torch._utils._rebuild_tensor_v2",
    • "torch.FloatStorage"

    What is a pickle import?

    123 MB
    xet
    add model about 4 years ago