https://github.com/autodistill/autodistill-yolov5
YOLOv5 module for use with Autodistill.
https://github.com/autodistill/autodistill-yolov5
autodistill computer-vision yolov5
Last synced: 2 months ago
JSON representation
YOLOv5 module for use with Autodistill.
- Host: GitHub
- URL: https://github.com/autodistill/autodistill-yolov5
- Owner: autodistill
- License: agpl-3.0
- Created: 2023-06-06T11:49:03.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-02-23T04:58:03.000Z (over 1 year ago)
- Last Synced: 2025-04-14T12:12:59.270Z (2 months ago)
- Topics: autodistill, computer-vision, yolov5
- Language: Python
- Homepage: https://docs.autodistill.com
- Size: 23.4 KB
- Stars: 1
- Watchers: 3
- Forks: 2
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Autodistill YOLOv5 Module
This repository contains the code supporting the YOLOv5 target model for use with [Autodistill](https://github.com/autodistill/autodistill).
[YOLOv5](https://github.com/ultralytics/ultralytics) is an open-source computer vision model by Ultralytics, the creators of YOLOv5. You can use `autodistill` to train a YOLOv5 object detection model on a dataset of labelled images generated by the base models that `autodistill` supports.
View our [YOLOv5 Instance Segmentation](/target-models/YOLOv5-instance-segmentation/) page for information on how to train instance segmentation models.
Read the full [Autodistill documentation](https://autodistill.github.io/autodistill/).
Read the [YOLOv5 Autodistill documentation](https://autodistill.github.io/autodistill/target_models/yolov5/).
## Installation
To use the YOLOv5 target model, you will need to install the following dependency:
```bash
pip3 install autodistill-yolov5
```
## Quickstart
```python
from autodistill_YOLOv5 import YOLOv5
target_model = YOLOv5("YOLOv5n.pt")
# train a model
target_model.train("./context_images_labeled/data.yaml", epochs=200)
# run inference on the new model
pred = target_model.predict("./context_images_labeled/train/images/dog-7.jpg", conf=0.01)
```
## License
The code in this repository is licensed under an [AGPL 3.0 license](LICENSE).
## 🏆 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!