Spaces:
Running
on
Zero
Running
on
Zero
Martin Tomov
commited on
bbox black to yellow
Browse files
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 =
|
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),
|