### 处理SUN RGB-D数据集的Python脚本 以下是用于加载和处理 SUN RGB-D 数据集中原图以及语言分割标签的一个 Python 脚本。此脚本假设用户已经下载并解压了 SUN RGB-D 数据集，并且该数据集包含了图像文件及其对应的标注信息。 #### 加载和预处理 SUN RGB-D 数据集 ```python import os from PIL import Image import numpy as np import json from tqdm import tqdm def load_sunrgbd_data(image_dir, label_dir): """ 加载 SUN RGB-D 数据集中的图片和对应的语言分割标签。 参数: image_dir (str): 原始图像所在的目录路径。 label_dir (str): 对应语言分割标签的 JSON 文件所在目录路径。 返回: list: 图像数组列表。 list: 标签字典列表。 """ images = [] labels = [] # 获取所有图像文件名 image_files = sorted(os.listdir(image_dir)) label_files = sorted([f for f in os.listdir(label_dir) if f.endswith('.json')]) # 遍历每张图像和其对应的标签文件 for img_file, lbl_file in tqdm(zip(image_files, label_files), desc="Processing"): try: # 加载原始图像 img_path = os.path.join(image_dir, img_file) img = np.array(Image.open(img_path).convert('RGB')) # 加载JSON格式的语义分割标签 lbl_path = os.path.join(label_dir, lbl_file) with open(lbl_path, 'r') as file: label_info = json.load(file) # 将图像和标签加入到列表中 images.append(img) labels.append(label_info) except Exception as e: print(f"Error processing {img_file}: {e}") return images, labels if __name__ == "__main__": # 替换为实际的数据集路径 IMAGE_DIR = "/path/to/sunrgbd/images" LABEL_DIR = "/path/to/sunrgbd/labels" if not os.path.exists(IMAGE_DIR) or not os.path.exists(LABEL_DIR): raise FileNotFoundError("指定的图像或标签目录不存在") # 加载数据 imgs, lbls = load_sunrgbd_data(IMAGE_DIR, LABEL_DIR) # 打印一些基本信息 print(f"成功加载 {len(imgs)} 张图像及对应的标签[^1]") ``` --- #### 关键说明 上述脚本实现了以下功能： 1. **读取图像**: 使用 `PIL.Image` 库加载 SUN RGB-D 的原始图像，将其转换为 NumPy 数组以便后续处理[^2]。 2. **解析标签**: 假设语言分割标签是以 JSON 格式存储的，通过标准库 `json` 解析这些标签文件的内容[^3]。 3. **错误处理**: 如果某些文件无法正常加载，则会跳过它们并打印警告消息[^4]。 需要注意的是，在运行此脚本之前，请确认已安装所需的依赖项（如 Pillow 和 tqdm），可以通过以下命令完成安装： ```bash pip install pillow tqdm ``` 此外，如果用户的 SUN RGB-D 数据集结构不同或者标签并非以 JSON 形式存在，则需调整相应部分来适配具体需求[^5]。 ---