使用Python生成20组 边界框(unique_boxes)数据集
发布时间:2023-12-11 04:24:40
生成 边界框数据集可以使用Python中的random模块和列表。边界框通常包含四个坐标值:x_min、y_min、x_max和y_max,代表边界框的左上角和右下角的坐标。以下是一个生成20个 边界框的例子:
import random
# 生成 边界框的函数
def generate_unique_boxes(num_boxes, min_value, max_value):
boxes = []
# 生成num_boxes个边界框
for _ in range(num_boxes):
# 生成 的坐标值
x_min = random.randint(min_value, max_value)
y_min = random.randint(min_value, max_value)
x_max = random.randint(x_min, max_value)
y_max = random.randint(y_min, max_value)
# 添加边界框到列表中
boxes.append((x_min, y_min, x_max, y_max))
return boxes
# 生成20个 边界框的数据集
unique_boxes = generate_unique_boxes(20, 0, 100)
# 打印生成的边界框
for i, box in enumerate(unique_boxes):
print(f"Box {i+1}: {box}")
运行上述代码,将会生成20个 的边界框,并且将其打印出来。每个边界框的坐标值都在0到100之间。输出结果类似于:
Box 1: (38, 22, 83, 83) Box 2: (92, 90, 96, 97) Box 3: (46, 45, 99, 80) ...
这样,你就生成了一个包含20个 边界框的数据集。你可以根据需要更改生成边界框的数量、坐标范围和格式。同时,你可以将这些边界框用于各种计算机视觉任务,比如目标检测、边界框回归等。
