experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix BMP header bytes.

Browse Source 
PiperOrigin-RevId: 317803857
Change-Id: Iafa8e4e44400ec3c119af8897ba3fa693138501f
This commit is contained in:
Hye Soo Yang 2020-06-22 22:38:05 -07:00 committed by TensorFlower Gardener
parent 934df8dcea
commit 088d437694
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/kernel_tests/decode_bmp_op_test.py
View File

@ -25,14 +25,14 @@ from tensorflow.python.ops import image_ops
from tensorflow.python.platform import test
class DecodeBmpOpTest(test.TestCase):
def testex1(self):
img_bytes = [[[0, 0, 255], [0, 255, 0]], [[255, 0, 0], [255, 255, 255]]]
# Encoded BMP bytes from Wikipedia
# BMP header bytes: https://en.wikipedia.org/wiki/List_of_file_signatures
encoded_bytes = [
0x42, 0x40,
0x42, 0x4d,
0x46, 0, 0, 0,
0, 0,
0, 0,
@ -66,9 +66,10 @@ class DecodeBmpOpTest(test.TestCase):
def testGrayscale(self):
img_bytes = [[[255], [0]], [[255], [0]]]
# BMP header bytes: https://en.wikipedia.org/wiki/List_of_file_signatures
encoded_bytes = [
0x42,
0x40,
0x4d,
0x3d,
0,
0,
@ -133,6 +134,8 @@ class DecodeBmpOpTest(test.TestCase):
byte_string = bytes(bytearray(encoded_bytes))
img_in = constant_op.constant(byte_string, dtype=dtypes.string)
# TODO(b/159600494): Currently, `decode_bmp` op does not validate input
# magic bytes.
decode = image_ops.decode_bmp(img_in)
with self.cached_session():