From c86d80a03180389aa4325ceb7ca7308e7a7d7359 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Wed, 21 Jun 2023 16:00:47 -0700 Subject: [PATCH] Internal Changes PiperOrigin-RevId: 542387813 --- mediapipe/model_maker/python/core/utils/BUILD | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/mediapipe/model_maker/python/core/utils/BUILD b/mediapipe/model_maker/python/core/utils/BUILD index 81bd68d3e..2c29970bb 100644 --- a/mediapipe/model_maker/python/core/utils/BUILD +++ b/mediapipe/model_maker/python/core/utils/BUILD @@ -80,6 +80,14 @@ py_test( deps = [":loss_functions"], ) +###################################################################### +# Public target of the MediaPipe Model Maker Quantization Config. + +# Quantization Config is used to export a quantized model. Please refer +# to the specific task documentations such as: +# https://developers.google.com/mediapipe/solutions/vision/image_classifier/customize +# for usage information. +###################################################################### py_library( name = "metrics", srcs = ["metrics.py"], @@ -95,6 +103,7 @@ py_library( name = "quantization", srcs = ["quantization.py"], srcs_version = "PY3", + visibility = ["//visibility:public"], deps = ["//mediapipe/model_maker/python/core/data:dataset"], )