遇见数据集

HuggingFaceM4/Stanford-Cars

收藏
Hugging Face2022-06-21 更新2024-03-04 收录
官方服务:

资源简介:

## 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: ![Sample with box position](data_readme/sample_100_box_pos.png)

用于可视化边界框(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() 结果:![带边界框的样本](data_readme/sample_100_box_pos.png)

提供机构:
HuggingFaceM4
原始信息汇总

数据集概述

数据集名称

  • 名称: Stanford-Cars
  • 来源: SaulLu

数据集加载与可视化

  • 加载方法: 使用load_dataset函数,通过指定数据集名称SaulLu/Stanford-Carsuse_auth_token=True进行加载。
  • 示例索引: 第100个样本
  • 样本信息: 包含图像文件路径和边界框坐标。
    • 图像路径: 从样本中提取的image.filename
    • 边界框坐标: 从样本中提取的bbox[0],包含四个坐标值。

可视化代码

  • 图像显示: 使用matplotlib.pyplot显示图像。
  • 边界框绘制: 使用matplotlib.patches.Rectangle在图像上绘制边界框,不填充。
  • 结果图像: 显示了第100个样本的图像及其边界框位置。
二维码
社区交流群
二维码
科研交流群
商业服务