Skip to content

Latest commit

 

History

History
77 lines (51 loc) · 3.48 KB

README.rst

File metadata and controls

77 lines (51 loc) · 3.48 KB

SAMYOL

License: Apache 2.0 GitHub stars lint-format-install

🤔 What is this?

SAMYOL is a Python library that combines an object detection model and a segmentation model. It provides a unified interface for performing object detection and segmentation tasks using different versions of the YOLO model (YOLOv6, YOLOv7, YOLOv8, or YOLO-NAS) and the Segment Anything Model (SAM).

🧩 Features

  • Integrated object detection and segmentation capabilities
  • Support for multiple versions of the YOLO model
  • Flexible input options for image paths
  • Easy-to-use interface for obtaining object detection predictions and segmentation masks
  • Supports SAM infrence from both Facebook and HuggingFace
Example 1 Example 3 Example 2
Mask 1 Mask 3 Mask 2

⏳ Quick Installation

SAMYOL is installed using pip.

This python library requires python>=3.10, as well as pytorch>=1.7.0 and torchvision>=0.8.1. Installing both PyTorch and TorchVision with CUDA support is strongly recommended. We will also need the models checkpoints. These will be installed automatically if they are not already installed.

pip install SAMYOL

🚀 Getting Started

The following notebook has detailed examples and usage instructions for each YOLO model:

💁 Contributing

As an open-source project in a rapidly developing field, we are extremely open to contributions, whether it be in the form of a new feature, improved infrastructure, or better documentation. For detailed information on how to contribute, see here