Eugene Siow
commited on
Commit
•
2ce8311
1
Parent(s):
8286bde
Add update to dataset Div2k reference.
Browse files- README.md +1 -2
- config.json +1 -2
README.md
CHANGED
|
@@ -4,7 +4,7 @@ tags:
|
|
| 4 |
- super-image
|
| 5 |
- image-super-resolution
|
| 6 |
datasets:
|
| 7 |
-
-
|
| 8 |
metrics:
|
| 9 |
- pnsr
|
| 10 |
- ssim
|
|
@@ -82,7 +82,6 @@ training_args = TrainingArguments(
|
|
| 82 |
|
| 83 |
config = EdsrConfig(
|
| 84 |
scale=4, # train a model to upscale 4x
|
| 85 |
-
supported_scales=[2, 3, 4],
|
| 86 |
)
|
| 87 |
model = EdsrModel(config)
|
| 88 |
|
|
|
|
| 4 |
- super-image
|
| 5 |
- image-super-resolution
|
| 6 |
datasets:
|
| 7 |
+
- eugenesiow/Div2k
|
| 8 |
metrics:
|
| 9 |
- pnsr
|
| 10 |
- ssim
|
|
|
|
| 82 |
|
| 83 |
config = EdsrConfig(
|
| 84 |
scale=4, # train a model to upscale 4x
|
|
|
|
| 85 |
)
|
| 86 |
model = EdsrModel(config)
|
| 87 |
|
config.json
CHANGED
|
@@ -6,6 +6,5 @@
|
|
| 6 |
"n_feats": 64,
|
| 7 |
"n_resblocks": 16,
|
| 8 |
"res_scale": 1,
|
| 9 |
-
"rgb_range": 255
|
| 10 |
-
"supported_scales": [2, 3, 4]
|
| 11 |
}
|
|
|
|
| 6 |
"n_feats": 64,
|
| 7 |
"n_resblocks": 16,
|
| 8 |
"res_scale": 1,
|
| 9 |
+
"rgb_range": 255
|
|
|
|
| 10 |
}
|