RashiAgarwal
commited on
Commit
·
5fd0532
1
Parent(s):
165b257
Update display.py
Browse files- 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) /
|
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.
|
86 |
0.7,
|
87 |
(0, 0, 0),
|
88 |
bbox_thick // 2,
|
89 |
-
lineType=cv2.
|
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
|