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

Update object_detector_test.py

Browse Source
This commit is contained in:
Kinar R 2023-04-15 00:29:33 +05:30 committed by GitHub
parent d6103b41bc
commit e5e8c71390
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/tasks/python/test/vision/object_detector_test.py
View File

@ -155,7 +155,7 @@ class ObjectDetectorTest(parameterized.TestCase):
) )
def _assert_approximately_equals_detection_result(self, result, expected_result): def _assert_approximately_equals_detection_result(self, result, expected_result):
self.assertLen(result.detections, 4) self.assertLen(result.detections, len(expected_result.detections))
for i, detection in enumerate(result.detections): for i, detection in enumerate(result.detections):
bbox = detection.bounding_box bbox = detection.bounding_box
category = detection.categories[0] category = detection.categories[0]