• Alara Dirik's avatar
    Adds image-guided object detection support to OWL-ViT (#20136) · a00b7e85
    Alara Dirik authored
    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