https://github.com/autodistill/autodistill-yolonas

YOLO-NAS module for use with Autodistill.
https://github.com/autodistill/autodistill-yolonas

autodistill computer-vision yolo-nas

Last synced: 5 days ago
JSON representation

YOLO-NAS module for use with Autodistill.

• Host: GitHub
• URL: https://github.com/autodistill/autodistill-yolonas
• Owner: autodistill
• License: other
• Created: 2023-06-06T14:29:27.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-02-12T11:25:45.000Z (9 months ago)
• Last Synced: 2024-09-16T12:44:53.128Z (about 2 months ago)
• Topics: autodistill, computer-vision, yolo-nas
• Language: Python
• Homepage: https://docs.autodistill.com
• Size: 19.5 KB
• Stars: 5
• Watchers: 3
• Forks: 1
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        


  


    

      

    

  



# Autodistill YOLO-NAS Module

This repository contains the code supporting the YOLO-NAS target model for use with [Autodistill](https://github.com/autodistill/autodistill).

[YOLO-NAS](https://github.com/Deci-AI/super-gradients/blob/master/YOLONAS.md) is an object detection model developed by [Deci AI](https://deci.ai/).

You can use `autodistill` to train a YOLO-NAS object detection model on a dataset of labelled images generated by the base models that `autodistill` supports.

Read the full [Autodistill documentation](https://autodistill.github.io/autodistill/).

Read the [YOLO-NAS Autodistill documentation](https://autodistill.github.io/autodistill/target_models/yolonas/).

## Installation

To use the YOLO-NAS target model, you will need to install the following dependency:

```bash

pip3 install autodistill-yolonas

```

## Quickstart

```python

from autodistill_yolonas import YOLONAS

target_model = YOLONAS()

# train a model

# specify the directory where your annotations (in YOLO format) are stored

target_model.train("./context_images_labeled", epochs=20)

# run inference on the new model

pred = target_model.predict("./context_images_labeled/train/images/dog-7.jpg", confidence=0.01)

```

## License

The YOLO-NAS model is licensed under the [YOLO-NAS License](https://github.com/Deci-AI/super-gradients/blob/master/LICENSE.YOLONAS.md).

## 🏆 Contributing

We love your input! Please see the core Autodistill [contributing guide](https://github.com/autodistill/autodistill/blob/main/CONTRIBUTING.md) to get started. Thank you 🙏 to all our contributors!