tests/models/owlvit/test_modeling_owlvit.py · a00b7e85ea4b3e3185440f1f82a6b58e3660b01d · chenpangpang / transformers

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

Alara Dirik authored Nov 16, 2022

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

a00b7e85

test_modeling_owlvit.py 29 KB

Replace test_modeling_owlvit.py