Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/ISSUE_TEMPLATE/bug-report.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ body:
attributes:
label: Search before asking
description: >
Please search the [YOLOE Docs](https://docs.ultralytics.com/models/yoloe/) and [issues](https://github.com/ultralytics/yoloe_data_engine/issues) to see if a similar bug report already exists.
Please search the [YOLOE Docs](https://docs.ultralytics.com/models/yoloe) and [issues](https://github.com/ultralytics/yoloe_data_engine/issues) to see if a similar bug report already exists.
options:
- label: >
I have searched the Ultralytics [issues](https://github.com/ultralytics/yoloe_data_engine/issues) and found no similar bug report.
Expand Down Expand Up @@ -74,7 +74,7 @@ body:
attributes:
label: Minimal Reproducible Example
description: >
When asking a question, people will be better able to provide help if you provide code that they can easily understand and use to **reproduce** the problem. This is referred to by community members as creating a [minimal reproducible example](https://docs.ultralytics.com/help/minimum-reproducible-example/).
When asking a question, people will be better able to provide help if you provide code that they can easily understand and use to **reproduce** the problem. This is referred to by community members as creating a [minimal reproducible example](https://docs.ultralytics.com/help/minimum-reproducible-example).
placeholder: |
```python
from data_engine import DataEngine
Expand Down
4 changes: 2 additions & 2 deletions .github/ISSUE_TEMPLATE/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,10 @@
blank_issues_enabled: true
contact_links:
- name: 📄 YOLOE Docs
url: https://docs.ultralytics.com/models/yoloe/
url: https://docs.ultralytics.com/models/yoloe
about: Ultralytics YOLOE model documentation
- name: 💬 Forum
url: https://community.ultralytics.com/
url: https://community.ultralytics.com
about: Ask on Ultralytics Community Forum
- name: 🎧 Discord
url: https://ultralytics.com/discord
Expand Down
2 changes: 1 addition & 1 deletion .github/ISSUE_TEMPLATE/feature-request.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ body:
attributes:
label: Search before asking
description: >
Please search the [YOLOE Docs](https://docs.ultralytics.com/models/yoloe/) and [issues](https://github.com/ultralytics/yoloe_data_engine/issues) to see if a similar feature request already exists.
Please search the [YOLOE Docs](https://docs.ultralytics.com/models/yoloe) and [issues](https://github.com/ultralytics/yoloe_data_engine/issues) to see if a similar feature request already exists.
options:
- label: >
I have searched the Ultralytics [issues](https://github.com/ultralytics/yoloe_data_engine/issues) and found no similar feature requests.
Expand Down
2 changes: 1 addition & 1 deletion .github/ISSUE_TEMPLATE/question.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ body:
attributes:
label: Search before asking
description: >
Please search the [YOLOE Docs](https://docs.ultralytics.com/models/yoloe/), [issues](https://github.com/ultralytics/yoloe_data_engine/issues) and [discussions](https://github.com/orgs/ultralytics/discussions) to see if a similar question already exists.
Please search the [YOLOE Docs](https://docs.ultralytics.com/models/yoloe), [issues](https://github.com/ultralytics/yoloe_data_engine/issues) and [discussions](https://github.com/orgs/ultralytics/discussions) to see if a similar question already exists.
options:
- label: >
I have searched the Ultralytics [issues](https://github.com/ultralytics/yoloe_data_engine/issues) and [discussions](https://github.com/orgs/ultralytics/discussions) and found no similar questions.
Expand Down
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,16 +1,16 @@
<a href="https://www.ultralytics.com/"><img src="https://raw.githubusercontent.com/ultralytics/assets/main/logo/Ultralytics_Logotype_Original.svg" width="320" alt="Ultralytics logo"></a>
<a href="https://www.ultralytics.com"><img src="https://raw.githubusercontent.com/ultralytics/assets/main/logo/Ultralytics_Logotype_Original.svg" width="320" alt="Ultralytics logo"></a>

# 🚀 YOLOE Data Engine

The YOLOE Data Engine builds and refines training data for [YOLOE](https://docs.ultralytics.com/models/yoloe/), the real-time open-vocabulary model in the [Ultralytics](https://www.ultralytics.com/) family. It closes the loop between a dataset and a model: existing labels are loaded, a trained YOLOE model predicts over the same images, and any prediction not already covered by a ground-truth box is merged back into the labels. The result is a denser, better-covered dataset for the next training run.
The YOLOE Data Engine builds and refines training data for [YOLOE](https://docs.ultralytics.com/models/yoloe), the real-time open-vocabulary model in the [Ultralytics](https://www.ultralytics.com) family. It closes the loop between a dataset and a model: existing labels are loaded, a trained YOLOE model predicts over the same images, and any prediction not already covered by a ground-truth box is merged back into the labels. The result is a denser, better-covered dataset for the next training run.

The pipeline handles both **grounding** data (free-form text phrases tied to image regions, such as Flickr30k Entities or mixed-grounding GQA) and **detection** data (a fixed class vocabulary, such as Objects365), and it includes tooling to refine the text prompts attached to grounding labels and to visually inspect everything it produces.

This is a working research repository: a collection of scripts rather than an installable package, with dataset, model, and buffer paths configured inside each script rather than passed on the command line.

[![Ultralytics Actions](https://github.com/ultralytics/yoloe_data_engine/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/yoloe_data_engine/actions/workflows/format.yml)
[![Ultralytics Discord](https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue)](https://discord.com/invite/ultralytics)
[![Ultralytics Forums](https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue)](https://community.ultralytics.com/)
[![Ultralytics Forums](https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue)](https://community.ultralytics.com)
[![Ultralytics Reddit](https://img.shields.io/reddit/subreddit-subscribers/ultralytics?style=flat&logo=reddit&logoColor=white&label=Reddit&color=blue)](https://reddit.com/r/ultralytics)

## 🔧 Requirements
Expand Down