Add cli for keras weight converter

Author: constantinpape

bioimageio/core/__main__.py

@@ -23,6 +23,11 @@
 except ImportError:
     torch_converter = None
 
+try:
+    from bioimageio.core.weight_converter import keras as keras_converter
+except ImportError:
+    keras_converter = None
+
 
 # extend help/version string by core version
 help_version_core = f"bioimageio.core {__version__}"
@@ -231,7 +236,8 @@ def convert_torch_weights_to_onnx(
         use_tracing: bool = typer.Option(True, help="Whether to use torch.jit tracing or scripting."),
         verbose: bool = typer.Option(True, help="Verbosity"),
     ) -> int:
-        return torch_converter.convert_weights_to_onnx(model_rdf, output_path, opset_version, use_tracing, verbose)
+        ret_code = torch_converter.convert_weights_to_onnx(model_rdf, output_path, opset_version, use_tracing, verbose)
+        sys.exit(ret_code)
 
     convert_torch_weights_to_onnx.__doc__ = torch_converter.convert_weights_to_onnx.__doc__
 
@@ -249,5 +255,21 @@ def convert_torch_weights_to_torchscript(
     convert_torch_weights_to_torchscript.__doc__ = torch_converter.convert_weights_to_pytorch_script.__doc__
 
 
+if keras_converter is not None:
+
+    @app.command()
+    def convert_keras_weights_to_tensorflow(
+        model_rdf: Path = typer.Argument(
+            ..., help="Path to the model resource description file (rdf.yaml) or zipped model."
+        ),
+        output_path: Path = typer.Argument(..., help="Where to save the tensorflow weights."),
+    ) -> int:
+        ret_code = keras_converter.convert_weights_to_tensorflow_saved_model_bundle(model_rdf, output_path)
+        sys.exit(ret_code)
+
+    convert_keras_weights_to_tensorflow.__doc__ =\
+        keras_converter.convert_weights_to_tensorflow_saved_model_bundle.__doc__
+
+
 if __name__ == "__main__":
     app()

bioimageio/core/weight_converter/keras/tensorflow.py

@@ -11,6 +11,8 @@
 from tensorflow import keras
 
 
+# adapted from
+# https://github.com/deepimagej/pydeepimagej/blob/master/pydeepimagej/yaml/create_config.py#L236
 def _convert_tf1(keras_weight_path, output_path, zip_weights):
     from tensorflow import saved_model
 

tests/test_cli.py

@@ -125,3 +125,10 @@ def test_torch_to_onnx(unet2d_nuclei_broad_model, tmp_path):
     ret = run_subprocess(["bioimageio", "convert-torch-weights-to-onnx", str(unet2d_nuclei_broad_model), str(out_path)])
     assert ret.returncode == 0, ret.stdout
     assert out_path.exists()
+
+
+def test_keras_to_tf(unet2d_keras, tmp_path):
+    out_path = tmp_path / "weights.zip"
+    ret = run_subprocess(["bioimageio", "convert-keras-weights-to-tensorflow", str(unet2d_keras), str(out_path)])
+    assert ret.returncode == 0, ret.stdout
+    assert out_path.exists()