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",
mediapipe/tasks/testdata/metadata/coco_ssd_mobilenet_v1_score_calibration.json
Vendored
+7 -14
View File
@@ -56,23 +56,20 @@
"max": 2
}
},
"stats": {
}
"stats": {}
},
{
"name": "category",
"description": "The categories of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
},
"content_properties": {},
"range": {
"min": 2,
"max": 2
}
},
"stats": {
},
"stats": {},
"associated_files": [
{
"name": "labels.txt",
@@ -86,8 +83,7 @@
"description": "The scores of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
},
"content_properties": {},
"range": {
"min": 2,
"max": 2
@@ -102,8 +98,7 @@
}
}
],
"stats": {
},
"stats": {},
"associated_files": [
{
"name": "score_calibration.txt",
@@ -117,11 +112,9 @@
"description": "The number of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
}
"content_properties": {}
},
"stats": {
}
"stats": {}
}
],
"output_tensor_groups": [
mediapipe/tasks/testdata/metadata/efficientdet_lite0_fp16_no_nms_anchors.csv
Vendored
+19206
View File
File diff suppressed because it is too large Load Diff
mediapipe/tasks/testdata/metadata/efficientdet_lite0_v1.json
Vendored
+7 -14
View File
@@ -42,15 +42,13 @@
"description": "The scores of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
},
"content_properties": {},
"range": {
"min": 2,
"max": 2
}
},
"stats": {
}
"stats": {}
},
{
"name": "location",
@@ -71,34 +69,29 @@
"max": 2
}
},
"stats": {
}
"stats": {}
},
{
"name": "number of detections",
"description": "The number of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
}
"content_properties": {}
},
"stats": {
}
"stats": {}
},
{
"name": "category",
"description": "The categories of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
},
"content_properties": {},
"range": {
"min": 2,
"max": 2
}
},
"stats": {
},
"stats": {},
"associated_files": [
{
"name": "labels.txt",
mediapipe/tasks/testdata/metadata/ssd_mobilenet_v1_no_metadata.json
Vendored
+7 -14
View File
@@ -56,23 +56,20 @@
"max": 2
}
},
"stats": {
}
"stats": {}
},
{
"name": "category",
"description": "The categories of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
},
"content_properties": {},
"range": {
"min": 2,
"max": 2
}
},
"stats": {
},
"stats": {},
"associated_files": [
{
"name": "labels.txt",
@@ -86,26 +83,22 @@
"description": "The scores of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
},
"content_properties": {},
"range": {
"min": 2,
"max": 2
}
},
"stats": {
}
"stats": {}
},
{
"name": "number of detections",
"description": "The number of the detected boxes.",
"content": {
"content_properties_type": "FeatureProperties",
"content_properties": {
}
"content_properties": {}
},
"stats": {
}
"stats": {}
}
],
"output_tensor_groups": [