1. 30 Nov, 2022 2 commits
    • amyeroberts's avatar
      Update doc examples feature extractor -> image processor (#20501) · 17a7b49b
      amyeroberts authored
      * Update doc example feature extractor -> image processor
      
      * Apply suggestions from code review
      17a7b49b
    • amyeroberts's avatar
      Add segmentation + object detection image processors (#20160) · de6d19ea
      amyeroberts authored
      * Add transforms for object detection
      
      * DETR models + Yolos
      
      * Scrappy additions
      
      * Maskformer image processor
      
      * Fix up; MaskFormer tests
      
      * Update owlvit processor
      
      * Add to docs
      
      * OwlViT tests
      
      * Update pad logic
      
      * Remove changes to transforms
      
      * Import fn directly
      
      * Update to include pad transformation
      
      * Remove uninstended changes
      
      * Add new owlvit post processing function
      
      * Tidy up
      
      * Fix copies
      
      * Fix some copies
      
      * Include device fix
      
      * Fix scipy imports
      
      * Update _pad_image
      
      * Update padding functionality
      
      * Fix bug
      
      * Properly handle ignore index
      
      * Fix up
      
      * Remove defaults to None in docstrings
      
      * Fix docstrings & docs
      
      * Fix sizes bug
      
      * Resolve conflicts in init
      
      * Cast to float after resizing
      
      * Tidy & add size if missing
      
      * Allow kwards when processing for owlvit
      
      * Update test values
      de6d19ea
  2. 18 Oct, 2022 1 commit
  3. 02 May, 2022 1 commit
    • NielsRogge's avatar
      Add YOLOS (#16848) · 1ac69874
      NielsRogge authored
      
      
      * First draft
      
      * Add YolosForObjectDetection
      
      * Make forward pass work
      
      * Add mid position embeddings
      
      * Add interpolation of position encodings
      
      * Add expected values
      
      * Add YOLOS to tests
      
      * Add integration test
      
      * Support tiny model as well
      
      * Support all models in conversion script
      
      * Remove mid_pe_size attribute
      
      * Make more tests pass
      
      * Add model to README and fix config
      
      * Add copied from statements
      
      * Rename base_model_prefix to vit
      
      * Add missing YOLOS_PRETRAINED_CONFIG_ARCHIVE_MAP
      
      * Apply suggestions from code review
      
      * Apply more suggestions from code review
      
      * Convert remaining checkpoints
      
      * Improve docstrings
      
      * Add YolosFeatureExtractor
      
      * Add feature extractor to docs
      
      * Add corresponding tests
      
      * Fix style
      
      * Fix docs
      
      * Apply suggestion from code review
      
      * Fix bad rebase
      
      * Fix some more bad rebase
      
      * Fix missing character
      
      * Improve docs and variable names
      Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
      1ac69874