跑yolov5代码想获得标注框的长宽比直方图

一鹿向晗99

已于 2023-05-18 15:02:14 修改

阅读量369

点赞数 1

文章标签： YOLO matplotlib python

于 2023-05-18 14:59:44 首次发布

本文链接：https://blog.csdn.net/qq_44133071/article/details/130747360

版权

该代码段使用Python的PIL和matplotlib库分析文本文件夹中的图像数据，计算并绘制标注框的长宽比直方图，用于理解图像对象的比例分布。程序读取txt文件中的坐标信息，计算比例并保存为直方图。

摘要由CSDN通过智能技术生成

结果：

代码：

import os
from PIL import Image
import matplotlib as mpl
import matplotlib.pyplot as plt
from matplotlib.pyplot import MultipleLocator
from matplotlib import ticker
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# txt文件夹路径
txts_path = "自己的路径"
# 图片文件夹路径
jpgs_path = "自己的路径"
# 存储w/h比例的列表
ratios = []
# 遍历文件夹中的每个txt文件
for txt_file in os.listdir(txts_path):
    if txt_file.endswith(".txt"):
        txt_path = os.path.join(txts_path, txt_file)
        txtname = os.path.basename(txt_path)
        jpgname = txtname.replace(".txt", ".jpg")
        jpg_path = os.path.join(jpgs_path, jpgname)
        image = Image.open(jpg_path)
        width, height = image.size
        with open(txt_path, "r") as file:
            lines = file.readlines()
            for line in lines:
                category, x, y, w, h = line.strip().split(" ")
                w = float(w)
                h = float(h)
                aw = w * width
                ah = h * height
                if ah > 0:
                    ratio = aw / ah
                    ratios.append(ratio)

# 生成直方图
plt.figure(figsize=(8, 5), dpi=80)
n, bins, patches = plt.hist(ratios, bins=10, rwidth=0.8, align='left')
# 打标签，在合适的位置标注每个直方图上面样本数
for i in range(len(n)):
    plt.text(bins[i], n[i]*1.02, int(n[i]), fontsize=12, horizontalalignment="center")
plt.ylim(0, 570)
# plt.legend()
plt.xlabel("Ratio (aw/ah)")
plt.ylabel("Frequency")
plt.title("标注框长宽比直方图")
plt.savefig('直方图'+'.png')
plt.show()
print('finish!')