你可以使用OpenCV库来可视化YOLO格式的数据集并绘制边界框。下面是一个简单的示例代码，可以帮助你完成这个任务：
```python
import cv2
import numpy as np
def visualize_dataset(image_path, label_path, class_names):
image = cv2.imread(image_path)
height, width, _ = image.shape
with open(label_path, 'r') as f:
lines = f.readlines()
for line in lines:
# 解析边界框信息
class_id, x, y, w, h = map(float, line.split())
class_id = int(class_id)
x = int((x - w/2) * width)
y = int((y - h/2) * height)
w = int(w * width)
h = int(h * height)
# 绘制边界框
cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)
cv2.putText(image, class_names[class_id], (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
cv2.imshow('Image', image)
cv2.waitKey(0)
cv2.destroyAllWindows()
# 设置类别名称和标签路径
class_names = ['person', 'car', 'cat'] # 替换成你的类别名称
label_path = 'path/to/labels.txt' # 替换成你的标签文件路径
image_path = 'path/to/image.jpg' # 替换成你的图像文件路径
# 可视化数据集
visualize_dataset(image_path, label_path, class_names)
```
在这个示例中，你需要将类别名称、标签文件路径和图像文件路径替换成你自己的路径。确保标签文件的格式与YOLO格式一致，每行包含了一个边界框的类别ID和边界框的中心坐标、宽度和高度。
运行这段代码后，你会看到一个显示了边界框的图像窗口。边界框使用绿色线条绘制在图像上，并显示了对应的类别名称。你可以使用键盘上的任意按键来关闭图像窗口。
希望这个示例能帮助到你！如果有任何疑问，请随时提问。
5126
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
