本文提供了一段Python代码，用于创建一个YOLOv5数据集的标注工具。该工具读取图像文件夹，允许用户通过鼠标绘制目标框并保存类别和位置信息，适用于图像标注任务。
首先，你需要准备好你的数据集图片和标签，标签通常是一个包含每个目标框的类别、位置和大小信息的文本文件。接下来，你可以使用如下 Python 代码来创建一个 YOLOv5 数据集标注工具：
```python import cv2 import numpy as np
class YOLOv5AnnotationTool: def init(self, image_folder, labels): self.image_folder = image_folder self.labels = labels self.current_image_index = 0 self.current_image_path = self.image_folder[self.current_image_index] self.current_image = cv2.imread(self.current_image_path) self.current_annotations = [] self.window_name = "YOLOv5 Annotation Tool" cv2.namedWindow(self.window_name) cv2.setMouseCallback(self.window_name, self.draw_bounding_box)
def draw_bounding_box(self, event, x, y, flags, param):
最低0.47元/天 解锁文章
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
