用Python生成包含20个 边界框(unique_boxes)的数据集
发布时间:2023-12-11 04:24:22
为了生成一个包含20个 边界框(unique_boxes)的数据集,我们可以使用Python的random模块来生成随机数。边界框通常表示为四个坐标:左上角的x和y坐标,以及宽度和高度。我们可以使用一个列表来保存这些边界框,并使用循环来生成每个 的边界框。
下面是一个生成包含20个 边界框的数据集的例子:
import random
def generate_unique_boxes_dataset(num_boxes):
unique_boxes = []
# 循环生成每个 的边界框
while len(unique_boxes) < num_boxes:
# 随机生成边界框的坐标和大小
x = random.randint(0, 100)
y = random.randint(0, 100)
width = random.randint(1, 10)
height = random.randint(1, 10)
# 创建一个边界框对象
box = {"x": x, "y": y, "width": width, "height": height}
# 检查边界框是否已经存在于列表中
if box not in unique_boxes:
# 添加边界框到列表中
unique_boxes.append(box)
return unique_boxes
# 使用示例:
dataset = generate_unique_boxes_dataset(20)
for i, box in enumerate(dataset):
print(f"边界框{i+1}:左上角坐标({box['x']},{box['y']}),宽度:{box['width']},高度:{box['height']}")
运行上述代码会生成一个包含20个 边界框的数据集,并打印出每个边界框的坐标、宽度和高度。注意,这个例子使用了随机数来生成边界框的坐标和大小,你可以根据自己的需求进行修改。
希望以上示例对你有帮助!
