在train_val.prototxt中使用ImageData类型字段时,GoogleNet无法读取图像

Question

我正在尝试使用caffe的GoogleNet实现.我想根据文本文件中的文件和标签列表训练深层网络,但问题是,当我训练深层网络时,它无法读取文件.

这是train_val.prototxt定义,我使用ImageData而不是使用具有'Data'类型的大型LMDB文件

name: "GoogleNet"
layer 
  {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  include {
  phase: TRAIN
 }
 transform_param 
 {
 mirror: true
 crop_size: 224
 mean_value: 104
 mean_value: 117
 mean_value: 123
 }
 data_param 
 {
  source: "path_to_file/file_paths_and_labels.txt"
  batch_size: 32
 }
}  

在这里我使用了ImageData类型用于googlenet,而不是像这里建议的那样输入数据:LMDB文件以及它们如何用于caffe深度学习网络

所以,我有文本文件(file_paths_and_labels.txt),其中每行包含以下内容:

path_to_image label

其中图像的路径是图像的地址,标签是图像的标签(有10个不同的标签).

我想确切地知道我错在哪里,因为当我运行深度网络训练命令时

./build/tools/caffe train --solver=/my_home/dl-caffe/models/bvlc_googlenet/solver.prototxt 

我有以下错误:

I0624 10:36:11.524896 15246 layer_factory.hpp:74] Creating layer data
I0624 10:36:11.524960 15246 net.cpp:84] Creating Layer data
I0624 10:36:11.524988 15246 net.cpp:338] data -> data
I0624 10:36:11.525046 15246 net.cpp:338] data -> label
I0624 10:36:11.525084 15246 net.cpp:113] Setting up data
I0624 10:36:11.525106 15246 image_data_layer.cpp:36] Opening file 
I0624 10:36:11.525146 15246 image_data_layer.cpp:51] A total of 0 images.
*** Aborted at 1435152971 (unix time) try "date -d @1435152971" if you   are using GNU date ***
PC: @     0x7f7060b70ee0 (unknown)
*** SIGSEGV (@0x0) received by PID 15246 (TID 0x7f706188aa40) from PID   0; stack trace: ***
@     0x7f7060511d40 (unknown)
@     0x7f7060b70ee0 (unknown)
@     0x7f706118587c std::operator+<>()
@     0x7f70611861e5 caffe::ImageDataLayer<>::DataLayerSetUp()
@     0x7f7061144ac6 caffe::BaseDataLayer<>::LayerSetUp()
@     0x7f7061144bc9 caffe::BasePrefetchingDataLayer<>::LayerSetUp()
@     0x7f70611d8ff2 caffe::Net<>::Init()
@     0x7f70611daab2 caffe::Net<>::Net()
@     0x7f70611e6c10 caffe::Solver<>::InitTrainNet()
@     0x7f70611e7d23 caffe::Solver<>::Init()
@     0x7f70611e7ef6 caffe::Solver<>::Solver()
@           0x40c4a0 caffe::GetSolver<>()
@           0x406471 train()
@           0x404a11 main
@     0x7f70604fcec5 (unknown)
@           0x404fbd (unknown)
Segmentation fault (core dumped)

我认为GoogleNet没有在我的文本文件中查找数据.问题是什么？我的train_val.prototxt文件的语法？

Answer 1

您使用错误的参数指定源.对于IMAGE_DATA,您需要使用image_data_param而不是data_param.因为您在data_param中指定了源,而ImageDataLayer查看了image_data_param,所以source的值是空字符串.您可以在此处的日志中看到:

I0624 10:36:11.525106 15246 image_data_layer.cpp:36] Opening file 

这一行的格式应该是:

Opening file <filename>

在您的日志中,"打开文件"后面有一个空白区域.