fix, from_florence_2: extra dimension for xyxy, REGION_TO_CATEGORY

This commit is contained in:
LinasKo committed 2024-06-20 21:35:49 +03:00
1 parent ec75a86a83
commit 23e63501eb
1 file changed
+1 -1
+1 -1
View File
@@ -177,7 +177,7 @@ def from_florence_2(
match is not None
), f"Expected string to end in location tags, but got {result}"
xyxy = np.array(match.groups(), dtype=np.float32)
xyxy = np.array([match.groups()], dtype=np.float32)
result_string = result[: match.start()]
labels = np.array([result_string])
return xyxy, labels, None, None