【绘图】将plt.imshow转换为cv2.imshow代码

1.本质来说就是一个画框函数的变换，改改就好了，另外注意的是plt.rectangle的坐标绘制依次为左上顶点坐标（x，y），w,h
cv2.rectangle的坐标为左上顶点坐标和右下顶点坐标。

2.其次plt是rgb方式读取。im = im[:, :, (2, 1, 0)] 对应着红，绿，蓝
cv2 imread读取图像之后，imshow是bgr通道顺序来进行显示的，如果使用pltshow去显示就会不正常。

也就是
bgr 0 1 2 opencv
rgb 2 1 0 plt

两段代码应用场景均是使用opencv去读取图像/视频帧。使用imread得到的图片帧（即im）通道为bgr

tf-faster-RCNN的代码

def vis_detections(im, class_name, dets, thresh=0.5):
    """Draw detected bounding boxes."""
    inds = np.where(dets[:, -1] >= thresh)[0]
    if len(inds) == 0:
        return

    im = im[:, :, (2, 1, 0)]    #因为用imread读入图像默认通道顺序为bgr，如果想要plt去显示，要转成rgb通道顺序
    fig, ax = plt.subplots(figsize=(12, 12))
    ax.imshow(im, aspect='equal')
    for i in inds:
        bbox = dets[i, :4]
        score = dets[i, -1]

        ax.add_patch(
            plt.Rectangle((bbox[0], bbox[1]),
                          bbox[2] - bbox[0],
                          bbox[3] - bbox[1], fill=False,
                          edgecolor='red', linewidth=3.5)
            )
        ax.text(bbox[0], bbox[1] - 2,
                '{:s} {:.3f}'.format(class_name, score),
                bbox=dict(facecolor='blue', alpha=0.5),
                fontsize=14, color='white')

    ax.set_title(('{} detections with '
                  'p({} | box) >= {:.1f}').format(class_name, class_name,
                                                  thresh),
                  fontsize=14)
    plt.axis('off')
    plt.tight_layout()
    plt.draw()

修改为opencv显示的代码：

   #这里注意的是，opencv imshow默认三通道顺序为bgr （imread读入的通道就是bgr，所以对应imshow也是bgr）
    def vis_detections(im, class_name, dets, thresh=0.5):
        """Draw detected bounding boxes."""
        inds = np.where(dets[:, -1] >= thresh)[0]
        if len(inds) == 0:
            return
    
        # im = im[:, :, (2, 1, 0)]
        for i in inds:
            bbox = dets[i, :4]
            score = dets[i, -1]
            cv2.rectangle(im,(bbox[0], bbox[1]),( bbox[2],bbox[3]),(0,255,255), 2)
            text='{:s} {:.3f}'.format(class_name, score)
            cv2.putText(im, text, (int(bbox[0]), int(bbox[1] - 2)), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,255,0), 2)

对通道数不懂得看这： https://blog.csdn.net/liu13364876993/article/details/79867061