【问题标题】:Error while training a basic network in caffe using ImageData layer使用 ImageData 层在 caffe 中训练基本网络时出错
【发布时间】:2018-07-21 21:33:38
【问题描述】:

我正在尝试开始使用 Caffe 训练网络进行图像分类。我希望求解器使用我自己的数据来训练模型。我正在使用"ImageData" layer 并准备了'file_lists.txt' 文件。 但我收到这些错误消息。

E0211 10:43:12.274907 60653568 io.cpp:80] Could not open or find file /my/path/img/myfirstimage.jpg 123
F0211 10:49:29.643215 146333696 image_data_layer.cpp:148] Check failed: cv_img.data Could not load /my/path/img/myfirstimage.jpg 123

'myfirstimage.jpg''file_lists.txt' 中的第一张图片,123 是标签。我检查了文件是否在路径'/my/path/img/myfirstimage.jpg'中。
它似乎正在尝试使用包含路径标签的字符串加载图像。通过查看文档和示例,除非我搞砸了,否则这不应该发生,但我什么也没看到。有人可以帮忙吗?

mynet.prototxt 的开头:

name: "mynet"
layer {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  image_data_param {
    source: "/my/path/file_list.txt"
    batch_size: 50
    new_height: 256
    new_width: 256
  }
  include {
      phase: TRAIN
  }
}
...

这里是mynet_solver.prototxt

net: "/my/path/mynet.prototxt"
test_iter: 10
test_interval: 500
base_lr: 0.01
momentum: 0.9
weight_decay: 0.0000
lr_policy: "inv"
gamma: 0.0001
power: 0.75
display: 100
max_iter: 50000
snapshot: 5000
snapshot_prefix: "/my/path"
solver_mode: CPU

我使用的命令:

caffe train -solver ./mynet_solver.prototxt

【问题讨论】:

    标签: neural-network computer-vision deep-learning caffe training-data


    【解决方案1】:

    您的诊断是正确的:似乎 caffe 正在尝试读取文件 "/my/path/img/myfirstimage.jpg 123",即路径标签。
    确保您的 '/my/path/file_list.txt' 具有这种确切格式:

    即:路径和标签之间只有一个空格(不是制表符或任何其他字符)。除了表示行尾的换行符之外,标签后没有其他空白字符。
    Caffe 使用c++ code 解析'/my/path/file_list.txt',因此解析不是很灵活。

    【讨论】:

      猜你喜欢
      • 2016-11-24
      • 1970-01-01
      • 2017-09-07
      • 2016-09-28
      • 2016-03-31
      • 2017-04-20
      • 2012-02-10
      • 2015-04-20
      • 2017-11-21
      相关资源
      最近更新 更多