From a650a8492860204f3a9987975a9b2d3ebec35306 Mon Sep 17 00:00:00 2001 From: nucli-vicky Date: Wed, 2 Sep 2026 09:37:49 +0200 Subject: [PATCH] add ffdnet --- .../solvers/ffdnet.py | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/ffdnet.py diff --git a/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/ffdnet.py b/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/ffdnet.py new file mode 100644 index 0000000..518a61c --- /dev/null +++ b/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/solvers/ffdnet.py @@ -0,0 +1,26 @@ +from benchopt import BaseSolver + +import torch +import deepinv as dinv + + +class Solver(BaseSolver): + name = "FFDNet" + + parameters = {} + + def set_objective(self, train_dataset=None, physics=None): + device = dinv.utils.get_freer_gpu() if torch.cuda.is_available() else "cpu" + self.model = dinv.models.ArtifactRemoval( + dinv.models.FFDNet( + n_conv_layers=12, nf=96, img_channels=3, norm=None, last_conv_bias=True + ), + device=device, + ) + self.model.device = device + + def run(self, _): + pass + + def get_result(self): + return dict(model=self.model)