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