make jpeg encoding more accurate

openAGI · May 10, 2024 · 900e785 · 900e785
1 parent 764921f
commit 900e785
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/datum/reader/parser.py b/datum/reader/parser.py
@@ -112,9 +112,9 @@ def decode_example(self, parsed_example: dict[str, tf.Tensor]) -> dict[str, tf.T
       dtype = self.pytype_to_tftype[self.datum_to_type_shape[key]['type']]
       if dtype == tf.string:
         if len(self.datum_to_type_shape[key]['shape']) == 2:
-          deserialized_outputs[key] = tf.io.decode_jpeg(value, 1)
+          deserialized_outputs[key] = tf.io.decode_jpeg(value, 1, dct_method="INTEGER_ACCURATE")
         elif len(self.datum_to_type_shape[key]['shape']) == 3:
-          deserialized_outputs[key] = tf.io.decode_jpeg(value, 3)
+          deserialized_outputs[key] = tf.io.decode_jpeg(value, 3, dct_method="INTEGER_ACCURATE")
         elif self.datum_to_type_shape[key]['dense']:
           deserialized_outputs[key] = value
         else:

diff --git a/datum/version.py b/datum/version.py
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '1.11.0'
+__version__ = '1.11.1'