• NielsRogge's avatar
    Add Swin2SR (#19784) · 26dd041c
    NielsRogge authored
    
    
    * First draft
    
    * Add more improvements
    
    * Improve forward pass
    
    * Fix layernorm
    
    * Add upscaler
    
    * More improvements
    
    * More improvements
    
    * More improvements
    
    * Improve conversion script
    
    * Add preprocessing
    
    * Make output match original implementation
    
    * Add additional attributes
    
    * Add support for more models
    
    * Support more models
    
    * Add support for real world sr
    
    * Add initial Swin2SRFeatureExtractor
    
    * Add ImageSuperResolutionOutput
    
    * Make more tests pass
    
    * Use BaseModelOutput
    
    * Fix one more test
    
    * Fix more tests
    
    * Fix another test
    
    * Fix all tests
    
    * Rename to Swin2SRImageProcessor
    
    * Fix toctree
    
    * Fix toctree
    
    * Fix rebase
    
    * Improve Swin2SRImageProcessor
    
    * Remove feature extractor file
    
    * Improve model
    
    * Improve conversion script
    
    * Fix integration test
    
    * Fix init
    
    * Fix conversion script
    
    * Address comments
    
    * Improve upsampler
    
    * Add NearestConvUpsampler
    
    * Improve pixel shuffle upsampler
    
    * Improve auxiliary upsampler
    
    * Improve conversion script
    
    * Rename conv_last to final_convolution
    
    * Fix rebase
    
    * Improve upsample module
    
    * Add padding to image processor
    
    * Fix bug
    
    * Update padding
    
    * Remove print statement and fix integration test
    
    * Improve docs
    
    * Add image processor tests
    
    * Convert all checkpoints, fix tests茅
    
    * Remove print statements
    
    * Fix import
    Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
    26dd041c
README_zh-hant.md 71.1 KB