Skip to content

Commit 6fcbbbd

Browse files
committed
fix gsdrunet + name of benchmark folder
1 parent 3177f4e commit 6fcbbbd

File tree

20 files changed

+5
-4
lines changed

20 files changed

+5
-4
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
**/__cache__/
44
**/data/
55
**/outputs/
6+
/tmp/**
67
**/*.egg-info/
78

89
# IDE config folders

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/plots

Lines changed: 0 additions & 1 deletion
This file was deleted.

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/config.yml renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/config.yml

File renamed without changes.

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/datasets/cbsd68.py renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/datasets/cbsd68.py

File renamed without changes.

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/objective.py renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/objective.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ class Objective(BaseObjective):
1212
"cbsd500_gaussian_denoising"
1313
)
1414

15-
requirements = ["pip::deepinv[dataset,denoisers,physics]", "datasets", "pip::pyiqa"]
15+
requirements = ["pip::deepinv[dataset,denoisers,physics]", "datasets"]
1616

1717
# Minimal version of benchopt required to run this benchmark.
1818
# Bump it up if the benchmark depends on a new feature of benchopt.
@@ -29,7 +29,7 @@ def evaluate_result(self, model):
2929
device = getattr(model, "device", None)
3030
self.physics = self.physics.to(device)
3131

32-
metrics = [dinv.loss.PSNR(), dinv.loss.NIQE(device=device)]
32+
metrics = [dinv.loss.PSNR()]
3333

3434
results = dinv.test(
3535
model,
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../plots/

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/solvers/bilateral_filter.py renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/bilateral_filter.py

File renamed without changes.

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/solvers/diffunet.py renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/diffunet.py

File renamed without changes.

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/solvers/dncnn.py renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/dncnn.py

File renamed without changes.

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/solvers/drunet.py renamed to deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/drunet.py

File renamed without changes.

0 commit comments

Comments
 (0)