test/generate.lua - master - valemobil/loadcaffe

test/generate.lua
68 строк2 KB
Sergey Zagoruyko
test update to support the update caffe binding
29 апр 2015, 12:06
29 апр 2015, 12:06c1c8122
require 'caffe'

nets_list = {}

table.insert(nets_list, {
  name = 'bvlc_alexnet',
  proto = './models/bvlc_alexnet/deploy.prototxt',
  binary = './models/bvlc_alexnet/bvlc_alexnet.caffemodel',
  imsize = 227,
  input = torch.rand(10,3,227,227):float(),
})

table.insert(nets_list, {
  name = 'bvlc_reference_caffenet',
  proto = './models/bvlc_reference_caffenet/deploy.prototxt',
  binary = './models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel',
  imsize = 227,
  input = torch.rand(10,3,227,227):float(),
})

table.insert(nets_list, {
  name = 'bvlc_reference_rcnn_ilsvrc13',
  proto = './models/bvlc_reference_rcnn_ilsvrc13/deploy.prototxt',
  binary = './models/bvlc_reference_rcnn_ilsvrc13/bvlc_reference_rcnn_ilsvrc13.caffemodel',
  imsize = 227,
  input = torch.rand(10,3,227,227):float(),
})
--[[
table.insert(nets_list, {
  name = 'finetune_flickr_style',
  proto = './models/034c6ac3865563b69e60/val.prototxt',
  binary = './models/034c6ac3865563b69e60/finetune_flickr_style.caffemodel',
  imsize = 227,
  input = torch.rand(10,3,227,227):float(),
})

table.insert(nets_list, {
  name = 'VGG_CNN_S',
  proto = './models/fd8800eeb36e276cd6f9/VGG_CNN_S_deploy.prototxt',
  binary = './models/fd8800eeb36e276cd6f9/VGG_CNN_S.caffemodel',
  imsize = 224,
  input = torch.rand(10,3,224,224):float()
})
--]]

table.insert(nets_list, {
  name = 'VGG_ILSVRC_19',
  proto = './models/3785162f95cd2d5fee77/VGG_ILSVRC_19_layers_deploy.prototxt',
  binary = './models/3785162f95cd2d5fee77/VGG_ILSVRC_19_layers.caffemodel',
  imsize = 224,
  input = torch.rand(10,3,224,224):float(),
})

table.insert(nets_list, {
  name = 'VGG_ILSVRC_16',
  proto = './models/211839e770f7b538e2d8/VGG_ILSVRC_16_layers_deploy.prototxt',
  binary = './models/211839e770f7b538e2d8/VGG_ILSVRC_16_layers.caffemodel',
  imsize = 224,
  input = torch.rand(10,3,224,224):float(),
})

for i,netprops in ipairs(nets_list) do
  net = caffe.Net(netprops.proto, netprops.binary, 'test')
  netprops.output = net:forward(netprops.input)
  net:reset()
end

torch.save('nets_list.bin', nets_list)