talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Metadata Writer: add object detection metadata writer.

Browse Source 
PiperOrigin-RevId: 513897494
This commit is contained in:
Yuqi Li
2023-03-06 07:11:10 -08:00
committed by Copybara-Service
parent 13db1c55d3
commit fe92d2e781
17 changed files with 1324 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/testdata/metadata/BUILD
Vendored
+20
View File
@@ -25,6 +25,12 @@ package(
mediapipe_files(srcs = [
"30k-clean.model",
"bert_text_classifier_no_metadata.tflite",
"category_tensor_float_meta.json",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29_no_metadata.tflite",
"coco_ssd_mobilenet_v1_score_calibration.json",
"efficientdet_lite0_v1.json",
"efficientdet_lite0_v1.tflite",
"labelmap.txt",
"mobile_ica_8bit-with-metadata.tflite",
"mobile_ica_8bit-with-unsupported-metadata-version.tflite",
"mobile_ica_8bit-without-model-metadata.tflite",
@@ -35,6 +41,10 @@ mediapipe_files(srcs = [
"mobilenet_v2_1.0_224_quant_without_metadata.tflite",
"mobilenet_v2_1.0_224_without_metadata.tflite",
"movie_review.tflite",
"score_calibration.csv",
"ssd_mobilenet_v1_no_metadata.json",
"ssd_mobilenet_v1_no_metadata.tflite",
"tensor_group_meta.json",
])
exports_files([
@@ -74,6 +84,8 @@ filegroup(
srcs = [
"30k-clean.model",
"bert_text_classifier_no_metadata.tflite",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29_no_metadata.tflite",
"efficientdet_lite0_v1.tflite",
"mobile_ica_8bit-with-metadata.tflite",
"mobile_ica_8bit-with-unsupported-metadata-version.tflite",
"mobile_ica_8bit-without-model-metadata.tflite",
@@ -83,6 +95,7 @@ filegroup(
"mobilenet_v2_1.0_224_quant_without_metadata.tflite",
"mobilenet_v2_1.0_224_without_metadata.tflite",
"movie_review.tflite",
"ssd_mobilenet_v1_no_metadata.tflite",
],
)
@@ -94,9 +107,12 @@ filegroup(
"bert_text_classifier_with_sentence_piece.json",
"bert_tokenizer_meta.json",
"bounding_box_tensor_meta.json",
"category_tensor_float_meta.json",
"classification_tensor_float_meta.json",
"classification_tensor_uint8_meta.json",
"classification_tensor_unsupported_meta.json",
"coco_ssd_mobilenet_v1_score_calibration.json",
"efficientdet_lite0_v1.json",
"external_file",
"feature_tensor_meta.json",
"general_meta.json",
@@ -107,6 +123,7 @@ filegroup(
"input_image_tensor_unsupported_meta.json",
"input_text_tensor_default_meta.json",
"input_text_tensor_meta.json",
"labelmap.txt",
"labels.txt",
"mobilebert_vocab.txt",
"mobilenet_v2_1.0_224.json",
@@ -114,10 +131,13 @@ filegroup(
"movie_review.json",
"movie_review_labels.txt",
"regex_vocab.txt",
"score_calibration.csv",
"score_calibration.txt",
"score_calibration_file_meta.json",
"score_calibration_tensor_meta.json",
"score_thresholding_meta.json",
"sentence_piece_tokenizer_meta.json",
"ssd_mobilenet_v1_no_metadata.json",
"tensor_group_meta.json",
],
)