Martin Tomov commited on
Commit
645e540
·
verified ·
1 Parent(s): 58e191c

bbox black to yellow

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
55
  score = detection.score
56
  box = detection.box
57
  mask = detection.mask
58
- color = np.random.randint(0, 256, size=3).tolist()
59
 
60
  cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, 2)
61
  cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),
 
55
  score = detection.score
56
  box = detection.box
57
  mask = detection.mask
58
+ color = (0, 255, 255) # Yellow in BGR format
59
 
60
  cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, 2)
61
  cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),