From 89e6b824aefa019154caef829f876f2ba7fbc446 Mon Sep 17 00:00:00 2001 From: Bekzhan Bekbolatuly Date: Fri, 14 Apr 2023 08:46:41 -0700 Subject: [PATCH] Update TF version to 2023-04-12 PiperOrigin-RevId: 524301262 --- WORKSPACE | 6 ++--- .../test/vision/object_detector_test.py | 22 ++++++++++++++----- third_party/org_tensorflow_custom_ops.diff | 6 ++--- 3 files changed, 23 insertions(+), 11 deletions(-) diff --git a/WORKSPACE b/WORKSPACE index fdea0a02c..6e079f142 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -458,9 +458,9 @@ http_archive( ) # TensorFlow repo should always go after the other external dependencies. -# TF on 2023-03-08. -_TENSORFLOW_GIT_COMMIT = "24f7ee636d62e1f8d8330357f8bbd65956dfb84d" -_TENSORFLOW_SHA256 = "7f8a96dd99215c0cdc77230d3dbce43e60102b64a89203ad04aa09b0a187a4bd" +# TF on 2023-04-12. +_TENSORFLOW_GIT_COMMIT = "d712c0c9e24519cc8cd3720279666720d1000eee" +_TENSORFLOW_SHA256 = "ba98de6ea5f720071246691a1536ecd5e1b1763033e8c82a1e721a06d3dfd4c1" http_archive( name = "org_tensorflow", urls = [ diff --git a/mediapipe/tasks/python/test/vision/object_detector_test.py b/mediapipe/tasks/python/test/vision/object_detector_test.py index 5a11246a4..8fdd81c17 100644 --- a/mediapipe/tasks/python/test/vision/object_detector_test.py +++ b/mediapipe/tasks/python/test/vision/object_detector_test.py @@ -49,7 +49,10 @@ _EXPECTED_DETECTION_RESULT = _DetectionResult( detections=[ _Detection( bounding_box=_BoundingBox( - origin_x=608, origin_y=161, width=381, height=439 + origin_x=608, + origin_y=164, + width=381, + height=432, ), categories=[ _Category( @@ -62,12 +65,15 @@ _EXPECTED_DETECTION_RESULT = _DetectionResult( ), _Detection( bounding_box=_BoundingBox( - origin_x=60, origin_y=398, width=386, height=196 + origin_x=57, + origin_y=398, + width=386, + height=196, ), categories=[ _Category( index=None, - score=0.64453125, + score=0.65625, display_name=None, category_name='cat', ) @@ -75,7 +81,10 @@ _EXPECTED_DETECTION_RESULT = _DetectionResult( ), _Detection( bounding_box=_BoundingBox( - origin_x=256, origin_y=395, width=173, height=202 + origin_x=256, + origin_y=394, + width=173, + height=202, ), categories=[ _Category( @@ -88,7 +97,10 @@ _EXPECTED_DETECTION_RESULT = _DetectionResult( ), _Detection( bounding_box=_BoundingBox( - origin_x=362, origin_y=191, width=325, height=419 + origin_x=360, + origin_y=195, + width=330, + height=412, ), categories=[ _Category( diff --git a/third_party/org_tensorflow_custom_ops.diff b/third_party/org_tensorflow_custom_ops.diff index 5de25a22e..9395faac2 100644 --- a/third_party/org_tensorflow_custom_ops.diff +++ b/third_party/org_tensorflow_custom_ops.diff @@ -3,9 +3,9 @@ index c49f2ce731d..d72773c0a5b 100644 --- a/tensorflow/lite/delegates/gpu/common/BUILD +++ b/tensorflow/lite/delegates/gpu/common/BUILD @@ -173,7 +173,7 @@ cc_library( - "//tensorflow/lite/kernels:kernel_util", - "//tensorflow/lite/kernels/internal:reference_base", - "//tensorflow/lite/kernels/internal:tensor", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/strings", - ] + tf_platform_alias("custom_parsers", "//tensorflow/lite/delegates/gpu/common/"), + ] + ["//tensorflow/lite/delegates/gpu/common/mediapipe:custom_parsers"], )