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()
|