windows下caffe训练自己的图片前期准备lmdb

最新推荐文章于 2023-07-01 10:25:51 发布

evanna-y

最新推荐文章于 2023-07-01 10:25:51 发布

阅读量3.2k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/sudakuang/article/details/52300992

已经接近崩溃状态……

8分类，准确率一直低于0.5并且准确率一直在震荡，从0升到0.5在退回去……

（一）准备图片

①lmdb/leveldb格式

a) 使用编译出来的convert_imageset.exe

D:\caffe\Build\x64\Release\convert_imageset.exe    D:\caffe\MyProject\lung\train\ train.txt  test_lmdb -backend=lmdb -resize_height=227 -resize_width=227 -shuffle
pause

注意txt与前面的数据文件夹中间有空格！！也就是说数据文件夹加上txt里的内容就是每张图片的全路径。txt的内容如下，注意换行应该用\n 而windows下用\r\n才会产生还行效果，\n在记事本中是显示不出来换行效果的不过不要紧，使用notepad打开就好了

b)直接使用图片不转换lmdb，运行会慢一点

txt 文件和上面差不多，只是要写全路径，保证程序读取txt文件就能找到张图片，这个需要使用的data层的type是imagedata，而不是data了

贴一下data层

layer {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    mirror: true
    crop_size: 224
    mean_file: "//tsclient/G/data/fintune/ResNet_mean.binaryproto"
  }
  image_data_param {
    source: "//tsclient/G/data/tra