Make dtype in Image class actually modifiable.
Changing dtype to any other type other than default will cause a crash because decode_jpeg or decode_image will promise to return uint8 all the time while decode_raw will actually vary its return type. This mismatch of types causes tf.case to fail and makes dtype parameter unusable. PiperOrigin-RevId: 210975290
Loading
Please sign in to comment