File tree Expand file tree Collapse file tree 4 files changed +5
-6
lines changed
bioimageio/core/weight_converters Expand file tree Collapse file tree 4 files changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -123,7 +123,7 @@ def _convert_tf2(
123
123
print ("TensorFlow model exported to" , output_path )
124
124
125
125
return TensorflowSavedModelBundleWeightsDescr (
126
- source = output_path ,
126
+ source = output_path . absolute () ,
127
127
parent = "keras_hdf5" ,
128
128
tensorflow_version = Version (tensorflow .__version__ ),
129
129
comment = f"Converted with bioimageio.core { __version__ } ." ,
@@ -138,7 +138,6 @@ def _convert_tf1(
138
138
input_name : str ,
139
139
output_name : str ,
140
140
) -> TensorflowSavedModelBundleWeightsDescr :
141
-
142
141
@no_type_check
143
142
def build_tf_model ():
144
143
keras_model = keras .models .load_model (keras_weight_path )
@@ -167,7 +166,7 @@ def build_tf_model():
167
166
print ("TensorFlow model exported to" , output_path )
168
167
169
168
return TensorflowSavedModelBundleWeightsDescr (
170
- source = output_path ,
169
+ source = output_path . absolute () ,
171
170
parent = "keras_hdf5" ,
172
171
tensorflow_version = Version (tensorflow .__version__ ),
173
172
comment = f"Converted with bioimageio.core { __version__ } ." ,
Original file line number Diff line number Diff line change @@ -72,7 +72,7 @@ def convert(
72
72
)
73
73
74
74
return OnnxWeightsDescr (
75
- source = output_path ,
75
+ source = output_path . absolute () ,
76
76
parent = "pytorch_state_dict" ,
77
77
opset_version = opset_version ,
78
78
comment = f"Converted with bioimageio.core { __version__ } ." ,
Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ def convert(
60
60
scripted_model .save (output_path )
61
61
62
62
return TorchscriptWeightsDescr (
63
- source = output_path ,
63
+ source = output_path . absolute () ,
64
64
pytorch_version = Version (torch .__version__ ),
65
65
parent = "pytorch_state_dict" ,
66
66
comment = (
Original file line number Diff line number Diff line change @@ -76,7 +76,7 @@ def convert(
76
76
)
77
77
78
78
return OnnxWeightsDescr (
79
- source = output_path ,
79
+ source = output_path . absolute () ,
80
80
parent = "torchscript" ,
81
81
opset_version = opset_version ,
82
82
comment = f"Converted with bioimageio.core { __version__ } ." ,
You can’t perform that action at this time.
0 commit comments