Combines YOLO models and SAM
Project description
======= SAMYOL
.. image:: https://img.shields.io/badge/License-Apache%202.0-blue.svg :target: https://opensource.org/licenses/Apache-2.0 :alt: License: Apache 2.0
.. image:: https://img.shields.io/github/stars/Jawher-Ben-Abdallah/SAMYOL.svg?style=social :alt: GitHub stars :target: https://github.com/Jawher-Ben-Abdallah/SAMYOL/stargazers
.. image:: https://github.com/Jawher-Ben-Abdallah/SAMYOL/actions/workflows/lint-format-install.yml/badge.svg :alt: lint-format-install :target: https://github.com/Jawher-Ben-Abdallah/SAMYOL/actions/workflows/lint-format-install.yml
🤔 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
.. raw:: html
.. raw:: html
⏳ Quick Installation
SAMYOL is installed using pip.
This python library requires python>=3.11, 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.
.. code:: python
Install SAMYOL:
pip install SAMYOL
or
conda install SAMYOL -c conda-forge
or
pip install git+https://github.com/Jawher-Ben-Abdallah/SAMYOL.git
or clone the repository locally and install
git clone git@github.com:Jawher-Ben-Abdallah/SAMYOL.git cd SAMYOL pip install .
🚀 Getting Started
The following notebook has detailed examples and usage instructions for each YOLO model:
.. image:: https://colab.research.google.com/assets/colab-badge.svg :target: https://colab.research.google.com/github/Jawher-Ben-Abdallah/SAMYOL/blob/main/SAMYOL.ipynb?authuser=1
💁 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 <https://github.com/Jawher-Ben-Abdallah/SAMYOL/blob/main/CONTRIBUTING.md>
_
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.