Skip to content

Implement YOLOE open-vocabulary instance segmentation model#34

Open
rycerzes wants to merge 2 commits intoCVHub520:mainfrom
rycerzes:yoloe
Open

Implement YOLOE open-vocabulary instance segmentation model#34
rycerzes wants to merge 2 commits intoCVHub520:mainfrom
rycerzes:yoloe

Conversation

@rycerzes
Copy link
Copy Markdown

Support YOLOE open-vocabulary instance segmentation models, including both text/visual prompt and prompt-free variants. This PR adds support for flexible segmentation modes via text prompts, visual prompts (bounding boxes), or prompt-free operation. The models are registered but commented out by default in the main models configuration.

YOLOE Segmentation Model Implementation:

  • Added a new YOLOESegmentation model class in app/models/yoloe.py supporting text prompt, visual prompt, and prompt-free segmentation modes, with detailed inference logic, mask-to-polygon conversion, and post-processing to output shapes.

Model Configuration Files:

Model Registration:

  • Registered multiple YOLOE variants (including 26n/s/m/l/x and v8 s/m/l, both prompt and prompt-free) with the @register_model decorator in yoloe.py, making them available for use in the platform.

Model Availability:

  • Updated configs/models.yaml to include (commented out) entries for the new YOLOE segmentation models, allowing easy activation as needed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant