RashiAgarwal commited on
Commit
5fd0532
·
1 Parent(s): 165b257

Update display.py

Browse files
Files changed (1) hide show
  1. display.py +3 -3
display.py CHANGED
@@ -50,7 +50,7 @@ def draw_predictions(image: np.ndarray, boxes: list[list], class_labels: list[st
50
 
51
  im = np.array(image)
52
  height, width, _ = im.shape
53
- bbox_thick = int((height + width) /1000)
54
 
55
  # Create a Rectangle patch
56
  for box in boxes:
@@ -82,11 +82,11 @@ def draw_predictions(image: np.ndarray, boxes: list[list], class_labels: list[st
82
  image,
83
  text,
84
  (x1, y1 - 2),
85
- cv2.FONT_ITALIC,
86
  0.7,
87
  (0, 0, 0),
88
  bbox_thick // 2,
89
- lineType=cv2.LINE_8,
90
  )
91
 
92
  return image
 
50
 
51
  im = np.array(image)
52
  height, width, _ = im.shape
53
+ bbox_thick = int((height + width) /500)
54
 
55
  # Create a Rectangle patch
56
  for box in boxes:
 
82
  image,
83
  text,
84
  (x1, y1 - 2),
85
+ cv2.FONT_HERSHEY_SIMPLEX,
86
  0.7,
87
  (0, 0, 0),
88
  bbox_thick // 2,
89
+ lineType=cv2.LINE_AA,
90
  )
91
 
92
  return image