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

Support hair segmentation model in image segmenter

Browse Source 
PiperOrigin-RevId: 515151150
This commit is contained in:
MediaPipe Team
2023-03-08 15:01:14 -08:00
committed by Copybara-Service
parent 2fb62e4c29
commit 9f1f4273d0
5 changed files with 246 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/testdata/vision/BUILD
Vendored
+8
View File
@@ -36,7 +36,9 @@ mediapipe_files(srcs = [
"coco_efficientdet_lite0_v1_1.0_quant_2021_09_06.tflite",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.tflite",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29_with_dummy_score_calibration.tflite",
"conv2d_input_channel_1.tflite",
"deeplabv3.tflite",
"dense.tflite",
"face_detection_full_range.tflite",
"face_detection_full_range_sparse.tflite",
"face_detection_short_range.tflite",
@@ -44,6 +46,7 @@ mediapipe_files(srcs = [
"face_landmark_with_attention.tflite",
"fist.jpg",
"fist.png",
"hair_segmentation.tflite",
"hand_landmark_full.tflite",
"hand_landmark_lite.tflite",
"hand_landmarker.task",
@@ -64,6 +67,7 @@ mediapipe_files(srcs = [
"pointing_up.jpg",
"pointing_up_rotated.jpg",
"portrait.jpg",
"portrait_hair_expected_mask.jpg",
"portrait_rotated.jpg",
"right_hands.jpg",
"right_hands_rotated.jpg",
@@ -117,6 +121,7 @@ filegroup(
"pointing_up.jpg",
"pointing_up_rotated.jpg",
"portrait.jpg",
"portrait_hair_expected_mask.jpg",
"portrait_rotated.jpg",
"right_hands.jpg",
"right_hands_rotated.jpg",
@@ -140,12 +145,15 @@ filegroup(
"coco_efficientdet_lite0_v1_1.0_quant_2021_09_06.tflite",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.tflite",
"coco_ssd_mobilenet_v1_1.0_quant_2018_06_29_with_dummy_score_calibration.tflite",
"conv2d_input_channel_1.tflite",
"deeplabv3.tflite",
"dense.tflite",
"face_detection_full_range.tflite",
"face_detection_full_range_sparse.tflite",
"face_detection_short_range.tflite",
"face_landmark.tflite",
"face_landmark_with_attention.tflite",
"hair_segmentation.tflite",
"hand_landmark_full.tflite",
"hand_landmark_lite.tflite",
"hand_landmarker.task",