用dummy数据层实现caffe time测试

命令:./build/tools/caffe time -iterations=10 --model=models/bvlc_alexnet/dummy_val.prototxt

网络:

修改原来网络数据层

name: "AlexNet"
layer {
  name: "data"
  type: "DummyData"
  top: "data"
#  top: "label"
#  include {
#    phase: TRAIN
#  }
#  transform_param {
#    mirror: true
#    crop_size: 227
#    mean_file: "data/ilsvrc12/imagenet_mean.binaryproto"
#  }
   dummy_data_param {
#    source: "examples/imagenet/ilsvrc12_train_lmdb"
#    batch_size: 256
#    backend: LMDB
     num: 256
     channels: 3
     height: 227#
     width: 227
   }
 }
 layer {
   name: "data"
   type: "DummyData"
 #  top: "data"
   top: "label"
 #  include {
 #    phase: TEST
 #  }
 #  transform_param {
 #    mirror: false
 #    crop_size: 227
 #    mean_file: "data/ilsvrc12/imagenet_mean.binaryproto"
 #  }
   dummy_data_param {
 #    source: "examples/imagenet/ilsvrc12_val_lmdb"
 #    batch_size: 50
 #    backend: LMDB
     num: 256
     channels: 1
     height: 1
     width: 1
   }
 }

posted @ 2016-08-02 16:55  venus024  阅读(1587)  评论(0)    收藏  举报