tensorflow
/
models
mirror of https://github.com/tensorflow/models.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

No public description 

PiperOrigin-RevId: 780592146
This commit is contained in:
Raviteja Gorijala 2025-07-08 09:24:35 -07:00 committed by A. Unique TensorFlower
parent f007603b50
commit d50c441ddd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
official/projects/volumetric_models/serving/semantic_segmentation_3d_test.py
View File

@ -60,9 +60,9 @@ class SemanticSegmentationExportTest(tf.test.TestCase, parameterized.TestCase):
image_tensor = tf.convert_to_tensor(self._image_array, dtype=tf.uint8)
return tf.expand_dims(image_tensor, axis=0)
if input_type == 'image_bytes':
return [self._image_array.tostring()]
return [self._image_array.tobytes()]
if input_type == 'tf_example':
encoded_image = self._image_array.tostring()
encoded_image = self._image_array.tobytes()
example = tf.train.Example(
features=tf.train.Features(
feature={