ultralytics / ultralytics

Ultralytics YOLO11 ๐Ÿš€
https://docs.ultralytics.com
GNU Affero General Public License v3.0
33.22k stars 6.39k forks source link

Text Contrastive Head in YOLO World #17747

Open toobatehreem opened 3 days ago

toobatehreem commented 3 days ago

Search before asking

Question

I am unable to find the code where the similarity score is calculated between object embeddings from YOLO and text embeddings from CLIP in YOLO World, please refer me to that code.

Additional

No response

UltralyticsAssistant commented 3 days ago

๐Ÿ‘‹ Hello @toobatehreem, thank you for your interest in Ultralytics ๐Ÿš€! We recommend checking out our comprehensive Documentation for a wealth of information on using our tools, including many helpful Python and CLI examples. There may be details in the documentation that directly relate to your question about implementation specifics.

If this is a ๐Ÿ› Bug Report, please ensure you provide a minimum reproducible example so we can better assist in debugging.

As your question pertains to specific functionality in YOLO World, having as much context as possible can helpโ€”consider providing any relevant logs or configurations you've been using.

Join our vibrant Ultralytics community to engage further:

Upgrade

To ensure you're using the latest features and fixes, upgrade to the newest ultralytics package with all requirements in a Python>=3.8 environment with PyTorch>=1.8:

pip install -U ultralytics

Environments

You can run YOLO in any of the following verified environments, each preloaded with necessary dependencies including CUDA/CUDNN, Python, and PyTorch:

Status

Ultralytics CI

If this badge is green, all of our Ultralytics CI checks are passing, confirming that the operational modes and tasks are functioning properly across macOS, Windows, and Ubuntu.

This is an automated response but rest assured, a member of our engineering team will review your query soon! ๐Ÿค–

Y-T-G commented 3 days ago

https://github.com/ultralytics/ultralytics/blob/cd6ef6105a3d43c8be70cd2d80aca4622d371a11/ultralytics/nn/modules/head.py#L315