matplotlib 画矩形的三种方式

假设矩形两点坐标如下，分别为：x1, y1, x2, y2

cat_dict['bbox'][i] = (min_row, min_col, max_row, max_col)

1. plt.plot(x,y)

这种方式画的矩形 因为边距的问题 会放缩

plt.plot([cat_dict['bbox'][i][1], cat_dict['bbox'][i][3], cat_dict['bbox'][i][3], cat_dict['bbox'][i][1], cat_dict['bbox'][i][1]],  # col
         [cat_dict['bbox'][i][0], cat_dict['bbox'][i][0], cat_dict['bbox'][i][2], cat_dict['bbox'][i][2], cat_dict['bbox'][i][0]],  # row
         color=[c / 255 for c in label_colors[cat_idx]], marker='.', ms=0)

2. ax.add_line(Line2D)

添加 4 条直线的方式，比较繁琐

from matplotlib.lines import Line2D

color = [c / 255 for c in label_colors[cat_idx]]
ax.add_line(Line2D([cat_dict['bbox'][i][1], cat_dict['bbox'][i][3]],
                   [cat_dict['bbox'][i][0], cat_dict['bbox'][i][0]], 
                   linewidth=2, 
                   color=color))
ax.add_line(Line2D([cat_dict['bbox'][i][3], cat_dict['bbox'][i][3]],
                   [cat_dict['bbox'][i][0], cat_dict['bbox'][i][2]], 
                   linewidth=2, 
                   color=color))
ax.add_line(Line2D([cat_dict['bbox'][i][3], cat_dict['bbox'][i][1]],
                   [cat_dict['bbox'][i][2], cat_dict['bbox'][i][2]], 
                   linewidth=2, 
                   color=color))
ax.add_line(Line2D([cat_dict['bbox'][i][1], cat_dict['bbox'][i][1]],
                   [cat_dict['bbox'][i][2], cat_dict['bbox'][i][0]], 
                   linewidth=2, 
                   color=color))

3. plt.gca().add_patch(plt.Rectangle())

最好的一种实现方式，fast rcnn 也是这么用的，传送门

plt.gca().add_patch(plt.Rectangle(xy=(cat_dict['bbox'][i][1], cat_dict['bbox'][i][0]),
                                  width=cat_dict['bbox'][i][3] - cat_dict['bbox'][i][1], 
                                  height=cat_dict['bbox'][i][2] - cat_dict['bbox'][i][0],
                                  edgecolor=[c / 255 for c in label_colors[cat_idx]],
                                  fill=False, linewidth=2))

效果图