使用matplotlib在图片上画框框

2024-06-17 12:24| 来源: 网络整理| 查看: 265

在物体检测任务中需要把检测出的物体用框框出来，下面就是用一个简单的例子展示如何使用matplotlib画框框

import matplotlib.pyplot as plt from PIL import Image import os root = "./data/coco/train2017" annFile = "./data/coco/annotations/instances_train2017.json" plt.imshow(Image.open(root + "/" + os.listdir(root)[0])) ax = plt.gca() # 默认框的颜色是黑色，第一个参数是左上角的点坐标 # 第二个参数是宽，第三个参数是长 ax.add_patch(plt.Rectangle((100, 200), 200, 100, color="blue", fill=False, linewidth=1)) # 第三个参数是标签的内容 # bbox里面facecolor是标签的颜色，alpha是标签的透明度 ax.text(100, 200, "label 0.9", bbox={'facecolor':'blue', 'alpha':0.5}) plt.savefig("./a.jpg") plt.show()

代码非常简单，其实就是用使用plt.gca()获取一个对象，然后在获取的对象里面添加矩形框和文本内容，具体每个参数的含义在注释中都有说明，下面就是最后的效果在这里插入图片描述

【本文地址】

公司简介

联系我们