Noah Vriese commited on
Commit
b70c559
·
1 Parent(s): 686c6f1

Update visualization formatting

Browse files
Files changed (1) hide show
  1. 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).capitalize()
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})"