banao-tech commited on
Commit
c776b64
·
verified ·
1 Parent(s): 17a86a0

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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 test.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()
 
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()