• NielsRogge's avatar
    Add DPT (#15991) · 979b039c
    NielsRogge authored
    
    
    * First draft
    
    * More improvements
    
    * Add fusion blocks
    
    * Make conversion script work for dpt_large
    
    * Make conversion script work
    
    * Improve implementation
    
    * Improve conversion script
    
    * Add DPTForSemanticSegmentation
    
    * Make conversion work for semantic segmentation
    
    * Add tests
    
    * Remove print statements
    
    * First draft
    
    * Redesign neck
    
    * Improve tests
    
    * Improve implementation some more
    
    * Make neck output list of tensors
    
    * Improve neck and feature extractor
    
    * Fix integration tests
    
    * Make more tests pass
    
    * Make all tests pass
    
    * Add missing config archive map
    
    * Add in_index attribute to make heads accept list of tensors
    
    * Apply suggestions from code review
    
    * Apply suggestions from code review
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    
    * Apply some more suggestions
    
    * Add copied from statements
    
    * Remove assert
    
    * Apply suggestions from code review
    
    * Apply suggestions from code review
    
    * Remove DPTInterpolate in favor of nn.Upsample
    
    * Add comments
    
    * Apply suggestions from code review
    
    * Apply suggestions from code review
    
    * Add proposed design
    
    * Update design
    
    * Add DPTReassembleLayer
    
    * Add DPTFeatureFusionStage
    
    * Apply more suggestions from code review
    
    * Apply suggestions from code review
    
    * Apply suggestions from code review
    
    * Fix rebase
    
    * Update in_index and out_indices
    
    * Fix conversion script
    
    * Fix code quality
    
    * Add model to toctree and use DepthEstimatorOutput
    
    * Fix rebase
    
    * Fix code examples
    
    * Improve code
    
    * Fix copied from statements
    
    * Apply suggestions from code review
    
    * Remove compute_loss method
    
    * Apply suggestions from code review
    
    * Fix documentation tests file
    
    * Remove test.py file
    
    * Improve doc example
    Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    Co-authored-by: default avatarNiels Rogge <nielsrogge@nielss-mbp.home>
    979b039c
README.md 53.3 KB