• NielsRogge's avatar
    Add SegFormer (#14019) · 1dc96a76
    NielsRogge authored
    
    
    * First draft
    
    * Make style & quality
    
    * Improve conversion script
    
    * Add print statement to see actual slice
    
    * Make absolute tolerance smaller
    
    * Fix image classification models
    
    * Add post_process_semantic method
    
    * Disable padding
    
    * Improve conversion script
    
    * Rename to ForSemanticSegmentation, add integration test, remove post_process methods
    
    * Improve docs
    
    * Fix code quality
    
    * Fix feature extractor tests
    
    * Fix tests for image classification model
    
    * Delete file
    
    * Add is_torch_available to feature extractor
    
    * Improve documentation of feature extractor methods
    
    * Apply suggestions from @sgugger's code review
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    
    * Apply some more suggestions of code review
    
    * Rebase with master
    
    * Fix rebase issues
    
    * Make sure model only outputs hidden states when the user wants to
    
    * Apply suggestions from code review
    
    * Add pad method
    
    * Support padding of 2d images
    
    * Add print statement
    
    * Add print statement
    
    * Move padding method to SegformerFeatureExtractor
    
    * Fix issue
    
    * Add casting of segmentation maps
    
    * Add test for padding
    
    * Add small note about padding
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    1dc96a76
README.md 43.8 KB