• Amrit Sahu's avatar
    [WIP] Add ZeroShotObjectDetectionPipeline (#18445) (#18930) · e9a49bab
    Amrit Sahu authored
    * Add ZeroShotObjectDetectionPipeline (#18445)
    
    * Add AutoModelForZeroShotObjectDetection task
    
    This commit also adds the following
    
    - Add explicit _processor method for ZeroShotObjectDetectionPipeline.
      This is necessary as pipelines don't auto infer processors yet and
      `OwlVitProcessor` wraps tokenizer and feature_extractor together, to
      process multiple images at once
    
    - Add auto tests and other tests for ZeroShotObjectDetectionPipeline
    
    * Add AutoModelForZeroShotObjectDetection task
    
    This commit also adds the following
    
    - Add explicit _processor method for ZeroShotObjectDetectionPipeline.
      This is necessary as pipelines don't auto infer processors yet and
      `OwlVitProcessor` wraps tokenizer and feature_extractor together, to
      process multiple images at once
    
    - Add auto tests and other tests for ZeroShotObjectDetectionPipeline
    
    * Add batching for ZeroShotObjectDetectionPipeline
    
    * Fix doc-string ZeroShotObjectDetectionPipeline
    
    * Fix output format: ZeroShotObjectDetectionPipeline
    e9a49bab
update_metadata.py 12 KB