cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated BUILD and tests

Browse Source
This commit is contained in:
kinaryml 2022-09-29 02:03:07 -07:00
parent d25626ff63
commit 63e0c04253
2 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mediapipe/tasks/python/test/vision/BUILD
View File

@ -26,7 +26,7 @@ py_test(
"//mediapipe/tasks/testdata/vision:test_models", "//mediapipe/tasks/testdata/vision:test_models",
], ],
deps = [ deps = [
# build rule placeholder: numpy dep, "//mediapipe/python:_framework_bindings",
"//mediapipe/tasks/python/components/containers:bounding_box", "//mediapipe/tasks/python/components/containers:bounding_box",
"//mediapipe/tasks/python/components/containers:category", "//mediapipe/tasks/python/components/containers:category",
"//mediapipe/tasks/python/components/containers:detections", "//mediapipe/tasks/python/components/containers:detections",
@ -34,7 +34,6 @@ py_test(
"//mediapipe/tasks/python/test:test_util", "//mediapipe/tasks/python/test:test_util",
"//mediapipe/tasks/python/vision:object_detector", "//mediapipe/tasks/python/vision:object_detector",
"//mediapipe/tasks/python/vision/core:vision_task_running_mode", "//mediapipe/tasks/python/vision/core:vision_task_running_mode",
"@absl_py//absl/testing:parameterized",
], ],
) )
@ -46,8 +45,7 @@ py_test(
"//mediapipe/tasks/testdata/vision:test_models", "//mediapipe/tasks/testdata/vision:test_models",
], ],
deps = [ deps = [
# build rule placeholder: numpy dep, "//mediapipe/python:_framework_bindings",
# build rule placeholder: cv2 dep,
"//mediapipe/tasks/python/core:base_options", "//mediapipe/tasks/python/core:base_options",
"//mediapipe/tasks/python/test:test_util", "//mediapipe/tasks/python/test:test_util",
"//mediapipe/tasks/python/components:segmenter_options", "//mediapipe/tasks/python/components:segmenter_options",

7
mediapipe/tasks/python/test/vision/image_segmenter_test.py
View File

@ -45,13 +45,6 @@ _MASK_MAGNIFICATION_FACTOR = 10
_MATCH_PIXELS_THRESHOLD = 0.01 _MATCH_PIXELS_THRESHOLD = 0.01
def _iou(ground_truth, prediction):
intersection = np.logical_and(ground_truth, prediction)
union = np.logical_or(ground_truth, prediction)
iou = np.sum(intersection) / np.sum(union)
return iou
class ModelFileType(enum.Enum): class ModelFileType(enum.Enum):
FILE_CONTENT = 1 FILE_CONTENT = 1
FILE_NAME = 2 FILE_NAME = 2