Rename flax data files

lanl · Jan 11, 2024 · 5fba865 · 5fba865
1 parent 52810bf
commit 5fba865
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/data b/data
diff --git a/scico/denoiser.py b/scico/denoiser.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-# Copyright (C) 2020-2023 by SCICO Developers
+# Copyright (C) 2020-2024 by SCICO Developers
 # All rights reserved. BSD 3-clause License.
 # This file is part of the SCICO package. Details of the copyright and
 # user license can be found in the 'LICENSE' file distributed with the
@@ -234,7 +234,7 @@ def __init__(self, variant: str = "6M"):
             self.is_blind = True
 
         model = DnCNNNet(depth=nlayer, channels=channels, num_filters=64, dtype=np.float32)
-        variables = load_weights(_flax_data_path("dncnn%s.npz" % variant))
+        variables = load_weights(_flax_data_path("dncnn%s.mpk" % variant))
         super().__init__(model, variables)
 
     def __call__(self, x: snp.Array, sigma: Optional[float] = None) -> snp.Array:
+2 −2		flax/README.rst
+ −		flax/dncnn17H.mpk
+ −		flax/dncnn17L.mpk
+ −		flax/dncnn17M.mpk
+ −		flax/dncnn17N.mpk
+ −		flax/dncnn6H.mpk
+ −		flax/dncnn6L.mpk
+ −		flax/dncnn6M.mpk
+ −		flax/dncnn6N.mpk