add unittest for dataset and archs

This commit is contained in:
Xintao
2021-11-28 15:59:14 +08:00
parent 7dd860a881
commit 1d180efaf3
18 changed files with 227 additions and 2 deletions

View File

@@ -0,0 +1,28 @@
name: Demo
type: RealESRGANDataset
dataroot_gt: tests/data/gt
meta_info: tests/data/meta_info_gt.txt
io_backend:
type: disk
blur_kernel_size: 21
kernel_list: ['iso', 'aniso', 'generalized_iso', 'generalized_aniso', 'plateau_iso', 'plateau_aniso']
kernel_prob: [0.45, 0.25, 0.12, 0.03, 0.12, 0.03]
sinc_prob: 1
blur_sigma: [0.2, 3]
betag_range: [0.5, 4]
betap_range: [1, 2]
blur_kernel_size2: 21
kernel_list2: ['iso', 'aniso', 'generalized_iso', 'generalized_aniso', 'plateau_iso', 'plateau_aniso']
kernel_prob2: [0.45, 0.25, 0.12, 0.03, 0.12, 0.03]
sinc_prob2: 1
blur_sigma2: [0.2, 1.5]
betag_range2: [0.5, 4]
betap_range2: [1, 2]
final_sinc_prob: 1
gt_size: 128
use_hflip: True
use_rot: False

View File

@@ -0,0 +1,13 @@
name: Demo
type: RealESRGANPairedDataset
scale: 4
dataroot_gt: tests/data
dataroot_lq: tests/data
meta_info: tests/data/meta_info_pair.txt
io_backend:
type: disk
phase: train
gt_size: 128
use_hflip: True
use_rot: False

BIN
tests/data/gt.lmdb/data.mdb Normal file

Binary file not shown.

BIN
tests/data/gt.lmdb/lock.mdb Normal file

Binary file not shown.

View File

@@ -0,0 +1,2 @@
baboon.png (480,500,3) 1
comic.png (360,240,3) 1

BIN
tests/data/gt/baboon.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 532 KiB

BIN
tests/data/gt/comic.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 195 KiB

BIN
tests/data/lq.lmdb/data.mdb Normal file

Binary file not shown.

BIN
tests/data/lq.lmdb/lock.mdb Normal file

Binary file not shown.

View File

@@ -0,0 +1,2 @@
baboon.png (120,125,3) 1
comic.png (80,60,3) 1

BIN
tests/data/lq/baboon.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 35 KiB

BIN
tests/data/lq/comic.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 14 KiB

View File

@@ -0,0 +1,2 @@
baboon.png
comic.png

View File

@@ -0,0 +1,2 @@
gt/baboon.png, lq/baboon.png
gt/comic.png, lq/comic.png