【发布时间】: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