# | Layer | Weight demension | stride |
---|---|---|---|
1 | conv | 32×32×5×5 | 2 |
2 | conv | 64×32×5×5 | 1 |
3 | conv | 64 × 64 × 5 × 5 | 2 |
4 | conv | 128×64×5×5 | 1 |
5 | conv | 128×128×5×5 | 4 |
6 | conv | 256×128×5×5 | 1 |
7 | conv | 256×256×5×5 | 4 |
8 | conv | 512×256×5×5 | 1 |
9 | conv | 512×512×4×4 | 4 |
10 | fc | 512×1 ×1 ×1 | - |
11 | sigmoid | - |
PSNR | SSIM | Runtime | |
---|---|---|---|
|
26.64 | 0.9142 | 0.93s |
|
27.33 | 0.9324 | 0.72s |