add unittest for dataset and archs
This commit is contained in:
28
tests/data/demo_option_realesrgan_dataset.yml
Normal file
28
tests/data/demo_option_realesrgan_dataset.yml
Normal 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
|
||||
13
tests/data/demo_option_realesrgan_paired_dataset.yml
Normal file
13
tests/data/demo_option_realesrgan_paired_dataset.yml
Normal 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
BIN
tests/data/gt.lmdb/data.mdb
Normal file
Binary file not shown.
BIN
tests/data/gt.lmdb/lock.mdb
Normal file
BIN
tests/data/gt.lmdb/lock.mdb
Normal file
Binary file not shown.
2
tests/data/gt.lmdb/meta_info.txt
Normal file
2
tests/data/gt.lmdb/meta_info.txt
Normal 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
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
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
BIN
tests/data/lq.lmdb/data.mdb
Normal file
Binary file not shown.
BIN
tests/data/lq.lmdb/lock.mdb
Normal file
BIN
tests/data/lq.lmdb/lock.mdb
Normal file
Binary file not shown.
2
tests/data/lq.lmdb/meta_info.txt
Normal file
2
tests/data/lq.lmdb/meta_info.txt
Normal 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
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
BIN
tests/data/lq/comic.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 14 KiB |
2
tests/data/meta_info_gt.txt
Normal file
2
tests/data/meta_info_gt.txt
Normal file
@@ -0,0 +1,2 @@
|
||||
baboon.png
|
||||
comic.png
|
||||
2
tests/data/meta_info_pair.txt
Normal file
2
tests/data/meta_info_pair.txt
Normal file
@@ -0,0 +1,2 @@
|
||||
gt/baboon.png, lq/baboon.png
|
||||
gt/comic.png, lq/comic.png
|
||||
Reference in New Issue
Block a user