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

Add custom metadata for object detection model with out-of-graph nms.

Browse Source 
PiperOrigin-RevId: 527083453
This commit is contained in:
MediaPipe Team
2023-04-25 14:58:51 -07:00
committed by Copybara-Service
parent 17f5b95387
commit 507ed0d91d
15 changed files with 19798 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/testdata/metadata/BUILD
Vendored
+5 -1
View File
@@ -32,7 +32,7 @@ mediapipe_files(srcs = [
"deeplabv3_with_activation.json",
"deeplabv3_without_labels.json",
"deeplabv3_without_metadata.tflite",
"efficientdet_lite0_v1.json",
"efficientdet_lite0_fp16_no_nms.tflite",
"efficientdet_lite0_v1.tflite",
"labelmap.txt",
"mobile_ica_8bit-with-custom-metadata.tflite",
@@ -64,6 +64,8 @@ exports_files([
"classification_tensor_float_meta.json",
"classification_tensor_uint8_meta.json",
"classification_tensor_unsupported_meta.json",
"efficientdet_lite0_fp16_no_nms_anchors.csv",
"efficientdet_lite0_fp16_no_nms.json",
"feature_tensor_meta.json",
"image_tensor_meta.json",
"input_image_tensor_float_meta.json",
@@ -94,6 +96,7 @@ filegroup(
"bert_text_classifier_no_metadata.tflite",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29_no_metadata.tflite",
"deeplabv3_without_metadata.tflite",
"efficientdet_lite0_fp16_no_nms.tflite",
"efficientdet_lite0_v1.tflite",
"mobile_ica_8bit-with-custom-metadata.tflite",
"mobile_ica_8bit-with-large-min-parser-version.tflite",
@@ -126,6 +129,7 @@ filegroup(
"deeplabv3.json",
"deeplabv3_with_activation.json",
"deeplabv3_without_labels.json",
"efficientdet_lite0_fp16_no_nms_anchors.csv",
"efficientdet_lite0_v1.json",
"external_file",
"feature_tensor_meta.json",