欢迎访问宙启技术站
智能推送

用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个 边界框的数据集,并打印出每个边界框的坐标、宽度和高度。注意,这个例子使用了随机数来生成边界框的坐标和大小,你可以根据自己的需求进行修改。

希望以上示例对你有帮助!