如何使用 Python 将图像和标签加载到 image_data_layer？答案

【问题标题】：How to load images and labels to image_data_layer using Python?如何使用 Python 将图像和标签加载到 image_data_layer？
【发布时间】：2016-11-16 20:14:10
【问题描述】：

我正在尝试将图像加载到网络并计算前 5 个错误率。我的数据层是

name: "CaffeNet"
layer {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  image_data_param {
      source: "/home/filipa/PycharmProjects/Proposal_Code/Labels/val.txt"

  }
}

其中 val.txt 文件包含每个图像的名称和对应的标签。（共 50000 张图片）

我的最后一层是

 name: "fc8"
  type: "InnerProduct"
  bottom: "fc7"
  top: "fc8"
  inner_product_param {
    num_output: 1000
  }
}
layer {
  name: "prob"
  type: "Softmax"
  bottom: "fc8"
  top: "prob"
}
layer {
  name: "accuracy_top_5"
  type: "Accuracy"
  bottom: "fc8"
  bottom: "label"
  top: "accuracy_top_5"
  accuracy_param {
    top_k: 5
  }
 include {
    phase: TEST
  }
}

我得到的错误是

E1116 11:34:28.173830  7465 io.cpp:80] Could not open or find file ILSVRC2012_val_00000001.JPEG
F1116 11:34:28.173835  7465 image_data_layer.cpp:72] Check failed: cv_img.data Could not load ILSVRC2012_val_00000001.JPEG

在我的代码中，我将图像加载到网络中

im = caffe.io.load_image(IMAGE_FILE)

IMAGE_FILE 指向我的图像的确切位置。

【问题讨论】：

标签： python neural-network caffe

【解决方案1】：

我会将 prototxt 编辑为：

  image_data_param {
      root_folder: "/home/filipa/PycharmProjects/root/folder/for/data"
      source: "/home/filipa/PycharmProjects/Proposal_Code/Labels/val.txt"
  }

然后将“val.txt”中的路径设置为相对于root_folder。

【讨论】：