1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
From: Bastian Germann <bage@debian.org>
Subject: Localize osam import
---
--- a/labelme/ai/text_to_annotation.py
+++ b/labelme/ai/text_to_annotation.py
@@ -2,13 +2,13 @@ import json
import time
import numpy as np
-import osam
from loguru import logger
def get_rectangles_from_texts(
model: str, image: np.ndarray, texts: list[str]
) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
+ import osam
request: osam.types.GenerateRequest = osam.types.GenerateRequest(
model=model,
image=image,
@@ -56,6 +56,7 @@ def non_maximum_suppression(
score_threshold: float,
max_num_detections: int,
) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
+ import osam
num_classes = np.max(labels) + 1
scores_of_all_classes = np.zeros((len(boxes), num_classes), dtype=np.float32)
for i, (score, label) in enumerate(zip(scores, labels)):
|
