Merge remote-tracking branch 'origin/feature/support_for_qwen_3_vl' into feature/support_for_qwen_3_vl

This commit is contained in:
SkalskiP committed 2025-11-15 21:37:05 +01:00
commit 353bd5336b
2 files changed
+12 -12

No files matched your search

+2 -2
View File
@@ -332,7 +332,7 @@ def from_qwen_2_5_vl(
start = text.find("[")
end = text.rfind("]")
if start != -1 and end != -1 and end > start:
text = text[start: end + 1].strip()
text = text[start : end + 1].strip()
try:
data = json.loads(text)
@@ -404,7 +404,7 @@ def from_qwen_3_vl(
result=result,
input_wh=(1000, 1000),
resolution_wh=resolution_wh,
classes=classes
classes=classes,
)
+10 -10
View File
@@ -320,7 +320,7 @@ def test_from_paligemma(
np.array(["dog"], dtype=str),
),
), # out-of-bounds box
(
(
does_not_raise(),
"""[
{'bbox_2d': [10, 20, 110, 120], 'label': 'cat'}
@@ -346,15 +346,15 @@ def test_from_paligemma(
(640, 640),
None,
(
np.array(
[
[0.0, 0.0, 64.0, 64.0],
[10.0, 20.0, 110.0, 120.0],
],
dtype=float,
),
None,
np.array(["dog", "cat"], dtype=str),
np.array(
[
[0.0, 0.0, 64.0, 64.0],
[10.0, 20.0, 110.0, 120.0],
],
dtype=float,
),
None,
np.array(["dog", "cat"], dtype=str),
),
), # truncated response, last object unfinished, previous ones recovered
(