Skip to content

GitLab

"server/vscode:/vscode.git/clone" did not exist on "f874c47831e12a1d447da12cb9e699cb2fb001da"
Commit de91fd22 authored by Gunho Park's avatar Gunho Park
Browse files

Internal change

parent ca0477a7
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
official/vision/beta/projects/basnet/serving/basnet.py
View file @ de91fd22
...@@ -60,6 +60,9 @@ class BASNetModule(semantic_segmentation.SegmentationModule): ...@@ -60,6 +60,9 @@ class BASNetModule(semantic_segmentation.SegmentationModule):
) )
masks = self.inference_step(images) masks = self.inference_step(images)
output = tf.image.resize(masks['ref'], self._input_image_size, method='bilinear') keys = sorted(masks.keys())
output = tf.image.resize(
masks[keys[-1]],
self._input_image_size, method='bilinear')
return dict(predicted_masks=output) return dict(predicted_masks=output)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment