Add Swin2SR#19784

Merged

NielsRogge merged 52 commits intohuggingface:mainfrom

NielsRogge:add_swin2sr

Dec 16, 2022

Commits on Dec 8, 2022

First draft

Niels Rogge
authored and
Niels Rogge
committed
Add more improvements

Niels Rogge
authored and
Niels Rogge
committed
Improve forward pass

Niels Rogge
authored and
Niels Rogge
committed
Fix layernorm

Niels Rogge
authored and
Niels Rogge
committed
Add upscaler

Niels Rogge
authored and
Niels Rogge
committed
More improvements

Niels Rogge
authored and
Niels Rogge
committed
More improvements

Niels Rogge
authored and
Niels Rogge
committed
More improvements

Niels Rogge
authored and
Niels Rogge
committed
Improve conversion script

Niels Rogge
authored and
Niels Rogge
committed
Add preprocessing

Niels Rogge
authored and
Niels Rogge
committed
Make output match original implementation

Niels Rogge
authored and
Niels Rogge
committed
Add additional attributes

Niels Rogge
authored and
Niels Rogge
committed
Add support for more models

Niels Rogge
authored and
Niels Rogge
committed
Support more models

Niels Rogge
authored and
Niels Rogge
committed
Add support for real world sr

Niels Rogge
authored and
Niels Rogge
committed
Add initial Swin2SRFeatureExtractor

Niels Rogge
authored and
Niels Rogge
committed
Add ImageSuperResolutionOutput

Niels Rogge
authored and
Niels Rogge
committed
Make more tests pass

Niels Rogge
authored and
Niels Rogge
committed
Use BaseModelOutput

Niels Rogge
authored and
Niels Rogge
committed
Fix one more test

Niels Rogge
authored and
Niels Rogge
committed
Fix more tests

Niels Rogge
authored and
Niels Rogge
committed
Fix another test

Niels Rogge
authored and
Niels Rogge
committed
Fix all tests

Niels Rogge
authored and
Niels Rogge
committed
Rename to Swin2SRImageProcessor

Niels Rogge
authored and
Niels Rogge
committed
Fix toctree

Niels Rogge
authored and
Niels Rogge
committed
Fix toctree

Niels Rogge
authored and
Niels Rogge
committed
Fix rebase

Niels Rogge
authored and
Niels Rogge
committed
Improve Swin2SRImageProcessor

Niels Rogge
authored and
Niels Rogge
committed
Remove feature extractor file

Niels Rogge
authored and
Niels Rogge
committed
Improve model

Niels Rogge
authored and
Niels Rogge
committed
Improve conversion script

Niels Rogge
authored and
Niels Rogge
committed
Fix integration test

Niels Rogge
authored and
Niels Rogge
committed
Fix init

Niels Rogge
authored and
Niels Rogge
committed
Fix conversion script

Niels Rogge
authored and
Niels Rogge
committed
Address comments

Niels Rogge
authored and
Niels Rogge
committed
Improve upsampler

Niels Rogge
authored and
Niels Rogge
committed
Add NearestConvUpsampler

Niels Rogge
authored and
Niels Rogge
committed
Improve pixel shuffle upsampler

Niels Rogge
authored and
Niels Rogge
committed
Improve auxiliary upsampler

Niels Rogge
authored and
Niels Rogge
committed
Improve conversion script

Niels Rogge
authored and
Niels Rogge
committed
Rename conv_last to final_convolution

Niels Rogge
authored and
Niels Rogge
committed
Fix rebase

Niels Rogge
authored and
Niels Rogge
committed
Improve upsample module

Niels Rogge
authored and
Niels Rogge
committed
Add padding to image processor

Niels Rogge
authored and
Niels Rogge
committed
Fix bug

Niels Rogge
authored and
Niels Rogge
committed

Commits on Dec 9, 2022

Update padding

Niels Rogge
authored and
Niels Rogge
committed
Remove print statement and fix integration test

Niels Rogge
authored and
Niels Rogge
committed

Commits on Dec 16, 2022

Improve docs

Niels Rogge
authored and
Niels Rogge
committed
Add image processor tests

Niels Rogge
authored and
Niels Rogge
committed
Convert all checkpoints, fix testsé

Niels Rogge
authored and
Niels Rogge
committed
Remove print statements

Niels Rogge
authored and
Niels Rogge
committed
Fix import

Niels Rogge
authored and
Niels Rogge
committed