Skip to content

Contributing to Supervision ๐Ÿ› ๏ธ

Thank you for your interest in contributing to Supervision!

We are actively improving this library to reduce the amount of work you need to do to solve common computer vision problems.

Contribution Guidelines

We welcome contributions to:

  1. Add a new feature to the library (guidance below).
  2. Improve our documentation and add examples to make it clear how to leverage the supervision library.
  3. Report bugs and issues in the project.
  4. Submit a request for a new feature.
  5. Improve our test coverage.

Contributing Features โœจ

Supervision is designed to provide generic utilities to solve problems. Thus, we focus on contributions that can have an impact on a wide range of projects.

For example, counting objects that cross a line anywhere on an image is a common problem in computer vision, but counting objects that cross a line 75% of the way through is less useful.

Before you contribute a new feature, consider submitting an Issue to discuss the feature so the community can weigh in and assist.

How to Contribute Changes

First, fork this repository to your own GitHub account. Click "fork" in the top corner of the supervision repository to get started:

Forking the repository

Creating a repository fork

Then, run git clone to download the project code to your computer.

Move to a new branch using the git checkout command:

git checkout -b <your_branch_name>

The name you choose for your branch should describe the change you want to make (i.e. line-counter-docs).

Make any changes you want to the project code, then run the following commands to commit your changes:

git add .
git commit -m "Your commit message"
git push -u origin main

๐ŸŽจ Code quality

Pre-commit tool

This project uses the pre-commit tool to maintain code quality and consistency. Before submitting a pull request or making any commits, it is important to run the pre-commit tool to ensure that your changes meet the project's guidelines.

Furthermore, we have integrated a pre-commit GitHub Action into our workflow. This means that with every pull request opened, the pre-commit checks will be automatically enforced, streamlining the code review process and ensuring that all contributions adhere to our quality standards.

To run the pre-commit tool, follow these steps:

  1. Install pre-commit by running the following command: poetry install. It will not only install pre-commit but also install all the deps and dev-deps of project

  2. Once pre-commit is installed, navigate to the project's root directory.

  3. Run the command pre-commit run --all-files. This will execute the pre-commit hooks configured for this project against the modified files. If any issues are found, the pre-commit tool will provide feedback on how to resolve them. Make the necessary changes and re-run the pre-commit command until all issues are resolved.

  4. You can also install pre-commit as a git hook by execute pre-commit install. Every time you made git commit pre-commit run automatically for you.

Docstrings

All new functions and classes in supervision should include docstrings. This is a prerequisite for any new functions and classes to be added to the library.

supervision adheres to the Google Python docstring style. Please refer to the style guide while writing docstrings for your contribution.

Type checking

So far, there is no type checking with mypy. See issue.

Then, go back to your fork of the supervision repository, click "Pull Requests", and click "New Pull Request".

Opening a pull request

Make sure the base branch is develop before submitting your PR.

On the next page, review your changes then click "Create pull request":

Configuring a pull request

Next, write a description for your pull request, and click "Create pull request" again to submit it for review:

Submitting a pull request

When creating new functions, please ensure you have the following:

  1. Docstrings for the function and all parameters.
  2. Unit tests for the function.
  3. Examples in the documentation for the function.
  4. Created an entry in our docs to autogenerate the documentation for the function.
  5. Please share a Google Colab with minimal code to test new feature or reproduce PR whenever it is possible. Please ensure that Google Colab can be accessed without any issue.

When you submit your Pull Request, you will be asked to sign a Contributor License Agreement (CLA) by the cla-assistant GitHub bot. We can only respond to PRs from contributors who have signed the project CLA.

All pull requests will be reviewed by the maintainers of the project. We will provide feedback and ask for changes if necessary.

PRs must pass all tests and linting requirements before they can be merged.

๐Ÿ“ documentation

The supervision documentation is stored in a folder called docs. The project documentation is built using mkdocs.

To run the documentation, install the project requirements with poetry install --with dev. Then, run mkdocs serve to start the documentation server.

You can learn more about mkdocs on the mkdocs website.

๐Ÿง‘โ€๐Ÿณ cookbooks

We are always looking for new examples and cookbooks to add to the supervision documentation. If you have a use case that you think would be helpful to others, please submit a PR with your example. Here are some guidelines for submitting a new example:

  • Create a new notebook in the docs/nodebooks folder.
  • Add a link to the new notebook in docs/theme/cookbooks.html. Make sure to add the path to the new notebook, as well as a title, labels, author and supervision version.
  • Use the Count Objects Crossing the Line example as a template for your new example.
  • Freeze the version of supervision you are using.
  • Place an appropriate Open in Colab button at the top of the notebook. You can find an example of such a button in the aforementioned Count Objects Crossing the Line cookbook.
  • Notebook should be self-contained. If you rely on external data ( videos, images, etc.) or libraries, include download and installation commands in the notebook.
  • Annotate the code with appropriate comments, including links to the documentation describing each of the tools you have used.

๐Ÿงช tests

pytests is used to run our tests.

๐Ÿ“„ license

By contributing, you agree that your contributions will be licensed under an MIT license.