Transformers
A2N
super-image
image-super-resolution
Inference Endpoints
Eugene Siow commited on
Commit
4d2c08a
1 Parent(s): 7490281

Add 2x model.

Browse files
Files changed (3) hide show
  1. README.md +4 -4
  2. config.json +5 -1
  3. pytorch_model_2x.pt +3 -0
README.md CHANGED
@@ -107,16 +107,16 @@ The results columns below are represented below as `PSNR/SSIM`. They are compare
107
 
108
  |Dataset |Scale |Bicubic |msrn-bam |
109
  |--- |--- |--- |--- |
110
- |Set5 |2x |33.64/0.9292 | |
111
  |Set5 |3x |30.39/0.8678 | |
112
  |Set5 |4x |28.42/0.8101 |**32.07/0.8933** |
113
- |Set14 |2x |30.22/0.8683 | |
114
  |Set14 |3x |27.53/0.7737 | |
115
  |Set14 |4x |25.99/0.7023 |**28.56/0.7801** |
116
- |BSD100 |2x |29.55/0.8425 | |
117
  |BSD100 |3x |27.20/0.7382 | |
118
  |BSD100 |4x |25.96/0.6672 |**27.54/0.7342** |
119
- |Urban100 |2x |26.66/0.8408 | |
120
  |Urban100 |3x | | |
121
  |Urban100 |4x |23.14/0.6573 |**25.89/0.7787** |
122
 
 
107
 
108
  |Dataset |Scale |Bicubic |msrn-bam |
109
  |--- |--- |--- |--- |
110
+ |Set5 |2x |33.64/0.9292 |**37.87/0.9602** |
111
  |Set5 |3x |30.39/0.8678 | |
112
  |Set5 |4x |28.42/0.8101 |**32.07/0.8933** |
113
+ |Set14 |2x |30.22/0.8683 |**33.45/0.9162** |
114
  |Set14 |3x |27.53/0.7737 | |
115
  |Set14 |4x |25.99/0.7023 |**28.56/0.7801** |
116
+ |BSD100 |2x |29.55/0.8425 |**32.11/0.8987** |
117
  |BSD100 |3x |27.20/0.7382 | |
118
  |BSD100 |4x |25.96/0.6672 |**27.54/0.7342** |
119
+ |Urban100 |2x |26.66/0.8408 |**31.71/0.9240** |
120
  |Urban100 |3x | | |
121
  |Urban100 |4x |23.14/0.6573 |**25.89/0.7787** |
122
 
config.json CHANGED
@@ -2,5 +2,9 @@
2
  "_name_or_path": "eugenesiow/a2n",
3
  "data_parallel": false,
4
  "model_type": "A2N",
5
- "supported_scales": [2, 3, 4]
 
 
 
 
6
  }
 
2
  "_name_or_path": "eugenesiow/a2n",
3
  "data_parallel": false,
4
  "model_type": "A2N",
5
+ "in_nc": 3,
6
+ "out_nc": 3,
7
+ "nf": 40,
8
+ "unf": 24,
9
+ "nb": 16
10
  }
pytorch_model_2x.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d74fae20445e10613abe91bc2a8c65dc4d5dff75adfc7d9b38476a242d174be
3
+ size 4212521