R语言新手求助：读取多文件夹图片用于图像处理的实现问题

嘿，作为R和图像处理的新手，你这个读取图片并添加额外列的需求我太懂啦！我帮你梳理了代码里的小问题，还整理了一套能直接用的方案，甚至考虑到了你后续要处理多文件夹的情况，一起来看看：

首先先修正你代码里的两个小细节：

• 路径要统一！你设置的工作目录是C:/Users/Ethan/Desktop/R，但list.files里写的路径是C:\\Users\\Desktop\\data\\images\\0，这里明显少了Ethan，肯定会找不到文件夹的。建议直接用绝对路径指定图片文件夹，避免工作目录的干扰。
• list.files的pattern参数要写严谨点，用"\\.jpg$"才能精准匹配后缀为jpg的文件，不然可能会误匹配到名字里带jpg但不是图片的文件；all.files参数一般默认FALSE就够，不用特意设置。

1. 加载工具包并设置正确路径 #

# 没装EBImage的话先跑install.packages("EBImage")
library(EBImage)

# 改成你实际的图片文件夹路径
img_dir <- "C:/Users/Ethan/Desktop/data/images/0"

2. 获取所有图片的文件名 #

# 获取文件夹下所有jpg图片，full.names=TRUE返回完整路径，方便直接读取
img_files <- list.files(path = img_dir, pattern = "\\.jpg$", full.names = TRUE)
# 提取纯文件名（不带路径），后续作为额外列内容
img_names <- basename(img_files)

3. 读取图片并整理成带额外列的数据框 #

EBImage读出来的图片是数组格式，我们可以把每张图片的像素值扁平化（转成一维向量），然后和你需要的额外信息（比如图片名、类别标签）一起组成数据框，方便后续训练：

# 初始化空列表存储单张图片的数据
img_data_list <- list()

# 循环遍历每一张图片
for (i in seq_along(img_files)) {
  # 读取当前图片
  img <- readImage(img_files[i])
  
  # 扁平化像素（要转灰度图的话加一句：img <- channel(img, "gray")）
  img_flat <- as.vector(img)
  
  # 创建带额外列的数据框，这里加了图片名和类别标签（假设该文件夹图片属于类别0）
  img_df <- data.frame(
    image_name = img_names[i],
    label = 0,  # 标签可根据实际需求调整
    t(img_flat)  # 转置让像素列横向排列，符合数据框结构
  )
  
  # 将当前图片数据加入列表
  img_data_list[[i]] <- img_df
}

# 合并所有图片数据为一个大的数据框
final_img_data <- do.call(rbind, img_data_list)

4. 检查结果 #

跑一下head(final_img_data)就能看到前几行数据，确认图片名、标签等额外列和像素列都正确加载了。

如果你后续要处理多个文件夹的图片（比如每个文件夹对应一个类别），只需要把逻辑改成循环遍历每个文件夹，自动给不同文件夹的图片加上对应标签就行：

# 假设所有类别文件夹都在这个根目录下
root_dir <- "C:/Users/Ethan/Desktop/data/images"
# 获取根目录下的所有子文件夹
class_folders <- list.dirs(root_dir, recursive = FALSE)

# 初始化空列表存储所有图片数据
total_data_list <- list()

# 循环遍历每个类别文件夹
for (j in seq_along(class_folders)) {
  current_dir <- class_folders[j]
  # 把文件夹名作为当前类别的标签
  current_label <- basename(current_dir)
  
  # 重复单文件夹的读取逻辑
  img_files <- list.files(path = current_dir, pattern = "\\.jpg$", full.names = TRUE)
  img_names <- basename(img_files)
  
  for (i in seq_along(img_files)) {
    img <- readImage(img_files[i])
    img_flat <- as.vector(img)
    img_df <- data.frame(
      image_name = img_names[i],
      label = as.integer(current_label),  # 将文件夹名转成整数标签
      t(img_flat)
    )
    total_data_list[[length(total_data_list)+1]] <- img_df
  }
}

# 合并所有文件夹的图片数据
final_total_data <- do.call(rbind, total_data_list)

这样就能批量处理所有文件夹的图片，自动给不同类别的图片加上对应标签，后续训练起来就方便多了～

内容的提问来源于stack exchange，提问作者Ethan