Skip to content

Commit 5d351fc

Browse files
Merge pull request #119 from bioimage-io/fix-pred
Fix inference with tf1
2 parents ee09857 + 80173fb commit 5d351fc

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bioimageio/core/prediction_pipeline/_model_adapters/_tensorflow_model_adapter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,8 @@ def _forward_tf(self, *input_tensors):
7272

7373
# run prediction
7474
res = sess.run(dict(zip(out_names, out_tensors)), dict(zip(in_tensors, input_tensors)))
75+
# from dict to list of tensors
76+
res = [res[out] for out in out_names]
7577

7678
return res
7779

0 commit comments

Comments
 (0)