463465810cz commited on
Commit ·
029f910
1
Parent(s): a95656a
update model
Browse filesFormer-commit-id: 0a843c2645645ca1c2e8e24ef4c020ebf2f14f08
- basicsr/models/dat_model.py +1 -1
- options/Test/test_DAT_2_x2.yml +1 -1
- options/Test/test_DAT_2_x3.yml +1 -1
- options/Test/test_DAT_2_x4.yml +1 -1
- options/Test/test_DAT_S_x2.yml +1 -1
- options/Test/test_DAT_S_x3.yml +1 -1
- options/Test/test_DAT_S_x4.yml +1 -1
- options/Test/test_DAT_light_x2.yml +1 -1
- options/Test/test_DAT_light_x3.yml +1 -1
- options/Test/test_DAT_light_x4.yml +1 -1
- options/Test/test_DAT_x2.yml +1 -1
- options/Test/test_DAT_x3.yml +1 -1
- options/Test/test_DAT_x4.yml +1 -1
- options/Test/test_single_x2.yml +1 -1
- options/Test/test_single_x3.yml +1 -1
- options/Test/test_single_x4.yml +1 -1
- options/Train/train_DAT_2_x2.yml +1 -1
- options/Train/train_DAT_2_x3.yml +1 -1
- options/Train/train_DAT_2_x4.yml +1 -1
- options/Train/train_DAT_S_x2.yml +1 -1
- options/Train/train_DAT_S_x3.yml +1 -1
- options/Train/train_DAT_S_x4.yml +1 -1
- options/Train/train_DAT_light_x2.yml +1 -1
- options/Train/train_DAT_light_x3.yml +1 -1
- options/Train/train_DAT_light_x4.yml +1 -1
- options/Train/train_DAT_x2.yml +1 -1
- options/Train/train_DAT_x3.yml +1 -1
- options/Train/train_DAT_x4.yml +1 -1
basicsr/models/dat_model.py
CHANGED
|
@@ -6,7 +6,7 @@ from basicsr.models.sr_model import SRModel
|
|
| 6 |
|
| 7 |
|
| 8 |
@MODEL_REGISTRY.register()
|
| 9 |
-
class
|
| 10 |
|
| 11 |
def test(self):
|
| 12 |
self.use_chop = self.opt['val']['use_chop'] if 'use_chop' in self.opt['val'] else False
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
@MODEL_REGISTRY.register()
|
| 9 |
+
class DATModel(SRModel):
|
| 10 |
|
| 11 |
def test(self):
|
| 12 |
self.use_chop = self.opt['val']['use_chop'] if 'use_chop' in self.opt['val'] else False
|
options/Test/test_DAT_2_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_2_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_2_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_2_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_2_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_2_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_2_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_2_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_2_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_S_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_S_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_S_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_S_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_S_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_S_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_S_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_S_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_S_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_light_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_light_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_light_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_light_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_light_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_light_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_light_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_light_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_light_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_DAT_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_DAT_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_single_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_single_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_single_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_single_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_single_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_single_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Test/test_single_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: test_single_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: test_single_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: 1
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_2_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_2_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_2_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_2_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_2_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_2_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_2_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_2_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_2_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_S_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_S_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_S_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_S_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_S_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_S_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_S_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_S_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_S_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_light_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_light_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_light_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_light_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_light_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_light_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_light_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_light_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_light_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_x2.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_x2
|
| 3 |
-
model_type:
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_x2
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 2
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_x3.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_x3
|
| 3 |
-
model_type:
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_x3
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 3
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
options/Train/train_DAT_x4.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_x4
|
| 3 |
-
model_type:
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|
|
|
|
| 1 |
# general settings
|
| 2 |
name: train_DAT_x4
|
| 3 |
+
model_type: DATModel
|
| 4 |
scale: 4
|
| 5 |
num_gpu: auto
|
| 6 |
manual_seed: 10
|