Add Fast owlvit Processor#37164
Conversation
|
Hi 👋, thank you for opening this pull request! The pull request is converted to draft by default. The CI will be paused while the PR is in draft mode. When it is ready for review, please click the |
yonigozlan
left a comment
yonigozlan
left a comment
There was a problem hiding this comment.
Thanks for contributing @keetrap ! Just a few things to change, otherwise LGTM
| image_mean = OPENAI_CLIP_MEAN | ||
| image_std = OPENAI_CLIP_STD | ||
| size = {"height": 768, "width": 768} | ||
| default_to_square = None |
There was a problem hiding this comment.
This should be default_to_square = True
It might also fix the issue with test_image_processor_from_dict_with_kwargs - AssertionError: {'shortest_edge': 42} != {'height': 42, 'width': 42}
There was a problem hiding this comment.
Thanks. Changes done
yonigozlan
left a comment
yonigozlan
left a comment
There was a problem hiding this comment.
Thanks for iterating @keetrap ! LGTM
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
* Add Fast Owlvit Processor * Update image_processing_owlvit_fast.py * Update image_processing_owlvit_fast.py --------- Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com>
* Add Fast Owlvit Processor * Update image_processing_owlvit_fast.py * Update image_processing_owlvit_fast.py --------- Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com>
3 participants
related #36978
cc @yonigozlan