SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Adds image-guided object detection support to OWL-ViT (#20136)

Browse Source 
Adds image-guided object detection method to OwlViTForObjectDetection class as described in the original paper. One-shot/ image-guided object detection enables users to use a query image to search for similar objects in the input image.

Co-Authored-By: Dhruv Karan k4r4n.dhruv@gmail.com
This commit is contained in:
Alara Dirik
2022-11-16 09:07:46 +03:00
committed by GitHub
parent 0d0d77693f
commit a00b7e85ea
7 changed files with 582 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/source/en/model_doc/owlvit.mdx
View File

@@ -80,6 +80,8 @@ This model was contributed by [adirik](https://huggingface.co/adirik). The origi
[[autodoc]] OwlViTFeatureExtractor
- __call__
- post_process
- post_process_image_guided_detection
## OwlViTProcessor
@@ -106,3 +108,4 @@ This model was contributed by [adirik](https://huggingface.co/adirik). The origi
[[autodoc]] OwlViTForObjectDetection
- forward
- image_guided_detection