Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Owlvit model enhancements #755

Merged
merged 13 commits into from
Oct 31, 2024
Merged

Owlvit model enhancements #755

merged 13 commits into from
Oct 31, 2024

Conversation

probicheaux
Copy link
Collaborator

@probicheaux probicheaux commented Oct 24, 2024

Description

Does a few things:

  1. Adds in negative prompting per class for owlvit
  2. Uses multiple examples per class by argmax, instead of meaning them
  3. Refactors some of these very heavy methods

Type of change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

Integration tests, and tested extensively via roboflow UI

Any specific deployment considerations

No

Docs

  • Docs updated? What were the changes:

Much of the thought here comes from @isaacrob-roboflow and https://github.com/DirectAI/simple-data-free-model-server

@probicheaux probicheaux marked this pull request as ready for review October 31, 2024 01:01
@probicheaux probicheaux changed the title Neg classes + concat instead of mean Owlvit model enhancements Oct 31, 2024
@PawelPeczek-Roboflow
Copy link
Collaborator

@probicheaux - shall we merge?

@probicheaux probicheaux merged commit ada5c76 into main Oct 31, 2024
57 checks passed
@probicheaux probicheaux deleted the owl-v2-improvements branch October 31, 2024 17:08
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.

2 participants