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

Metadata Writer: Add Metadata Writer for image classifier.

Browse Source 
PiperOrigin-RevId: 483282627
This commit is contained in:
Yuqi Li
2022-10-23 23:05:32 -07:00
committed by Copybara-Service
parent ec2a34d2a4
commit ab17be9294
11 changed files with 1537 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/testdata/metadata/BUILD
Vendored
+10
View File
@@ -29,6 +29,8 @@ mediapipe_files(srcs = [
"mobile_object_classifier_v0_2_3-metadata-no-name.tflite",
"mobilenet_v1_0.25_224_1_default_1.tflite",
"mobilenet_v2_1.0_224_quant.tflite",
"mobilenet_v2_1.0_224_quant_without_metadata.tflite",
"mobilenet_v2_1.0_224_without_metadata.tflite",
])
exports_files([
@@ -48,6 +50,9 @@ exports_files([
"score_calibration.txt",
"score_calibration_file_meta.json",
"score_calibration_tensor_meta.json",
"labels.txt",
"mobilenet_v2_1.0_224.json",
"mobilenet_v2_1.0_224_quant.json",
])
filegroup(
@@ -59,6 +64,8 @@ filegroup(
"mobile_object_classifier_v0_2_3-metadata-no-name.tflite",
"mobilenet_v1_0.25_224_1_default_1.tflite",
"mobilenet_v2_1.0_224_quant.tflite",
"mobilenet_v2_1.0_224_quant_without_metadata.tflite",
"mobilenet_v2_1.0_224_without_metadata.tflite",
],
)
@@ -78,6 +85,9 @@ filegroup(
"input_image_tensor_float_meta.json",
"input_image_tensor_uint8_meta.json",
"input_image_tensor_unsupported_meta.json",
"labels.txt",
"mobilenet_v2_1.0_224.json",
"mobilenet_v2_1.0_224_quant.json",
"score_calibration.txt",
"score_calibration_file_meta.json",
"score_calibration_tensor_meta.json",