Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -381,7 +381,7 @@ def annotate(image_source: np.ndarray, boxes: torch.Tensor, logits: torch.Tensor
|
|
| 381 |
labels = [f"{phrase}" for phrase in range(boxes.shape[0])]
|
| 382 |
|
| 383 |
# Import the custom box annotator from your project structure.
|
| 384 |
-
from
|
| 385 |
box_annotator = BoxAnnotator(text_scale=text_scale, text_padding=text_padding,
|
| 386 |
text_thickness=text_thickness, thickness=thickness)
|
| 387 |
annotated_frame = image_source.copy()
|
|
|
|
| 381 |
labels = [f"{phrase}" for phrase in range(boxes.shape[0])]
|
| 382 |
|
| 383 |
# Import the custom box annotator from your project structure.
|
| 384 |
+
from util.box_annotator import BoxAnnotator
|
| 385 |
box_annotator = BoxAnnotator(text_scale=text_scale, text_padding=text_padding,
|
| 386 |
text_thickness=text_thickness, thickness=thickness)
|
| 387 |
annotated_frame = image_source.copy()
|