HuggingFaceM4/Stanford-Cars
收藏资源简介:
## Code snippet to visualise the position of the box ```python import matplotlib.image as img import matplotlib.pyplot as plt from datasets import load_dataset from matplotlib.patches import Rectangle # Load dataset ds_name = "SaulLu/Stanford-Cars" ds = load_dataset(ds_name, use_auth_token=True) # Extract information for the sample we want to show index = 100 sample = ds["train"][index] box_coord = sample["bbox"][0] img_path = sample["image"].filename # Create plot # define Matplotlib figure and axis fig, ax = plt.subplots() # plot figure image = img.imread(img_path) ax.imshow(image) # add rectangle to plot ax.add_patch( Rectangle((box_coord[2], box_coord[0]), box_coord[3] - box_coord[2], box_coord[1] - box_coord[0], fill=None) ) # display plot plt.show() ``` Result: 
用于可视化边界框(bounding box)位置的代码片段: python import matplotlib.image as img import matplotlib.pyplot as plt from datasets import load_dataset from matplotlib.patches import Rectangle # 加载数据集 ds_name = "SaulLu/Stanford-Cars" ds = load_dataset(ds_name, use_auth_token=True) # 提取待展示样本的相关信息 index = 100 sample = ds["train"][index] box_coord = sample["bbox"][0] # 提取该样本的首个边界框坐标 img_path = sample["image"].filename # 创建绘图对象 # 定义Matplotlib的画布与坐标轴 fig, ax = plt.subplots() # 绘制图像 image = img.imread(img_path) ax.imshow(image) # 向绘图中添加矩形框 ax.add_patch( Rectangle((box_coord[2], box_coord[0]), box_coord[3] - box_coord[2], box_coord[1] - box_coord[0], fill=None) ) # 显示绘图结果 plt.show() 结果:
数据集概述
数据集名称
- 名称: Stanford-Cars
- 来源: SaulLu
数据集加载与可视化
- 加载方法: 使用
load_dataset函数,通过指定数据集名称SaulLu/Stanford-Cars和use_auth_token=True进行加载。 - 示例索引: 第100个样本
- 样本信息: 包含图像文件路径和边界框坐标。
- 图像路径: 从样本中提取的
image.filename。 - 边界框坐标: 从样本中提取的
bbox[0],包含四个坐标值。
- 图像路径: 从样本中提取的
可视化代码
- 图像显示: 使用
matplotlib.pyplot显示图像。 - 边界框绘制: 使用
matplotlib.patches.Rectangle在图像上绘制边界框,不填充。 - 结果图像: 显示了第100个样本的图像及其边界框位置。



