jpxkqx commited on
Commit
f37b77b
1 Parent(s): a6d9ce4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -187,7 +187,7 @@ The model architecture is based on the original Swin2 arcthitecture for Super Re
187
  ![architecture](architecture.png)
188
 
189
  The main component of the model is a [transformers.Swin2SRModel](https://huggingface.co/docs/transformers/model_doc/swin2sr#transformers.Swin2SRModel) which increases x8 the spatial resolution of its inputs (Swin2SR only supports upscaling ratios power of 2).
190
- As the real upscale ratio is ~5 and the output shape of the region considered is (160, 240), a Convolutional Neural Network (CNN) is included as a pre-process component which convert the inputs into a (40, 60) feature maps that can be fed to the Swin2SRModel.
191
 
192
  This network is trained to learn the residuals of the bicubic interpolation.
193
 
 
187
  ![architecture](architecture.png)
188
 
189
  The main component of the model is a [transformers.Swin2SRModel](https://huggingface.co/docs/transformers/model_doc/swin2sr#transformers.Swin2SRModel) which increases x8 the spatial resolution of its inputs (Swin2SR only supports upscaling ratios power of 2).
190
+ As the real upscale ratio is ~5 and the output shape of the region considered is (160, 240), a Convolutional Neural Network (CNN) is included as a pre-process component which convert the inputs into a (20, 30) feature maps that can be fed to the Swin2SRModel.
191
 
192
  This network is trained to learn the residuals of the bicubic interpolation.
193