Noah Vriese
commited on
Commit
·
b70c559
1
Parent(s):
686c6f1
Update visualization formatting
Browse files- data_objects.py +4 -1
data_objects.py
CHANGED
@@ -202,11 +202,14 @@ class Detection:
|
|
202 |
|
203 |
if draw_text:
|
204 |
|
205 |
-
display_text: str = str(self.display_name)
|
206 |
text_position: Tuple[int, int] = (
|
207 |
centroid_coords[0] + text_position_xy[0],
|
208 |
centroid_coords[1] + text_position_xy[1]
|
209 |
)
|
|
|
|
|
|
|
210 |
|
211 |
if hasattr(self, "status") and self.status:
|
212 |
display_text += f" ({self.status})"
|
|
|
202 |
|
203 |
if draw_text:
|
204 |
|
205 |
+
display_text: str = str(self.display_name)
|
206 |
text_position: Tuple[int, int] = (
|
207 |
centroid_coords[0] + text_position_xy[0],
|
208 |
centroid_coords[1] + text_position_xy[1]
|
209 |
)
|
210 |
+
|
211 |
+
if hasattr(self, "score") and self.score:
|
212 |
+
display_text += f" ({self.score})"
|
213 |
|
214 |
if hasattr(self, "status") and self.status:
|
215 |
display_text += f" ({self.status})"
|