Add update to dataset Div2k reference.

README.md

@@ -4,7 +4,7 @@ tags:
 - super-image
 - image-super-resolution
 datasets:
-- div2k
+- eugenesiow/Div2k
 metrics:
 - pnsr
 - ssim
@@ -14,7 +14,7 @@ MSRN model pre-trained on DIV2K (800 images training, augmented to 4000 images,
 
 The goal of image super resolution is to restore a high resolution (HR) image from a single low resolution (LR) image. The image below shows the ground truth (HR), the bicubic upscaling x2 and model upscaling x2.
 
-![Comparing Bicubic upscaling against the models x2 upscaling on Set5 Image 4](images/msrn_4_4_compare.png "Comparing Bicubic upscaling against the models x2 upscaling on Set5 Image 4")
+![Comparing Bicubic upscaling against the models x4 upscaling on Set5 Image 4](images/msrn_4_4_compare.png "Comparing Bicubic upscaling against the models x4 upscaling on Set5 Image 4")
 ## Model description
 The MSRN model proposes a feature extraction structure called the multi-scale residual block. This module can "adaptively detect image features at different scales" and "exploit the potential features of the image".
 
@@ -83,7 +83,6 @@ training_args = TrainingArguments(
 config = MsrnConfig(
     scale=4,                                # train a model to upscale 4x
     bam=True,                               # apply balanced attention to the network
-    supported_scales=[2, 3, 4],
 )
 model = MsrnModel(config)
 
@@ -122,7 +121,7 @@ The results columns below are represented below as `PSNR/SSIM`. They are compare
 |Urban100  	    |3x  	    |  	                |**29.31/0.8737**  	    |
 |Urban100  	    |4x  	    |23.14/0.6573  	    |**26.10/0.7857**  	    |
 
-![Comparing Bicubic upscaling against the models x2 upscaling on Set5 Image 2](images/msrn_2_4_compare.png "Comparing Bicubic upscaling against the models x2 upscaling on Set5 Image 2")
+![Comparing Bicubic upscaling against the models x4 upscaling on Set5 Image 2](images/msrn_2_4_compare.png "Comparing Bicubic upscaling against the models x4 upscaling on Set5 Image 2")
 
 ## BibTeX entry and citation info
 ```bibtex

config.json

@@ -5,6 +5,5 @@
   "bam": true,
   "n_feats": 64,
   "n_blocks": 8,
-  "rgb_range": 255,
-  "supported_scales": [2,3,4]
+  "rgb_range": 255
 }