Skip to content

Commit 14a61d6

Browse files
committed
fix axis on object detection tutorial
1 parent 93d08ea commit 14a61d6

File tree

1 file changed

+2
-0
lines changed
  • machine-learning/object-detection

1 file changed

+2
-0
lines changed

machine-learning/object-detection/utils.py

+2
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ def nms(boxes, iou_thresh):
5757
"""
5858
Performs Non maximal suppression technique to `boxes` using `iou_thresh` threshold
5959
"""
60+
# print(boxes.shape)
6061
# If there are no bounding boxes do nothing
6162
if len(boxes) == 0:
6263
return boxes
@@ -261,5 +262,6 @@ def get_color(c, x, max_val):
261262
a.text(x1 + lxc, y1 - lyc, conf_tx, fontsize = 12, color = 'k',
262263
bbox = dict(facecolor = rgb, edgecolor = rgb, alpha = 0.6))
263264

265+
plt.axis("off")
264266
plt.savefig("output.jpg")
265267
plt.show()

0 commit comments

Comments
 (0)