xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix florence 2 tests

This commit is contained in:
LinasKo 2024-07-05 13:21:59 +03:00
parent d3db15fcd0
commit a8b12914c8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test/detection/test_lmm_florence_2.py
View File

@ -237,7 +237,7 @@ from supervision.detection.lmm import from_florence_2
DoesNotRaise(),
),
( # Region to Category: detected
{"<REGION_TO_CATEGORY>": "some object<loc_3><loc_4><loc_5><loc_6>"},
{"<REGION_TO_CATEGORY>": "some object<loc_300><loc_400><loc_500><loc_600>"},
(10, 10),
(
np.array([[3, 4, 5, 6]], dtype=np.float32),
@ -254,11 +254,11 @@ from supervision.detection.lmm import from_florence_2
DoesNotRaise(),
),
( # Region to Description: detected
{"<REGION_TO_DESCRIPTION>": "some description<loc_3><loc_4><loc_5><loc_6>"},
{"<REGION_TO_DESCRIPTION>": "descr<loc_300><loc_400><loc_500><loc_600>"},
(10, 10),
(
np.array([[3, 4, 5, 6]], dtype=np.float32),
np.array(["some description"]),
np.array(["descr"]),
None,
None,
),